def from_dict(cls, d: Dict) -> 'StructureGraph': if 'nn_strategy' in d: nn_strategy = d.pop('nn_strategy') nn_strategy_obj = local_env.deserialize(nn_strategy) d.update({'nn_strategy': nn_strategy_obj}) return super().from_dict(d) return super().from_dict(d)
def test_serialization(self): mall = MinimumDistanceNNAll(4) config = serialize(mall) self.assertDictEqual( config, {"@module": "megnet.data.local_env", "@class": "MinimumDistanceNNAll", "cutoff": 4} ) self.assertTrue(serialize(None) is None) mall2 = deserialize(config) self.assertTrue(isinstance(mall2, MinimumDistanceNNAll)) self.assertTrue(mall2.cutoff == 4)
def test_serialization(self): mall = MinimumDistanceNNAll(4) config = serialize(mall) self.assertDictEqual( config, { '@module': 'megnet.data.local_env', '@class': 'MinimumDistanceNNAll', 'cutoff': 4 }) self.assertTrue(serialize(None) is None) mall2 = deserialize(config) self.assertTrue(isinstance(mall2, MinimumDistanceNNAll)) self.assertTrue(mall2.cutoff == 4)
def from_dict(cls, d: Dict) -> "StructureGraph": """ Initialization from dictionary Args: d (dict): dictionary Returns: StructureGraph object """ if "nn_strategy" in d: nn_strategy = d.pop("nn_strategy") nn_strategy_obj = local_env.deserialize(nn_strategy) d.update({"nn_strategy": nn_strategy_obj}) return super().from_dict(d) return super().from_dict(d)