diff --git a/m3gnet/graph/_converters.py b/m3gnet/graph/_converters.py index aa12a44..ac5afee 100644 --- a/m3gnet/graph/_converters.py +++ b/m3gnet/graph/_converters.py @@ -110,7 +110,7 @@ def __call__(self, structure: StructureOrMolecule, state_attr=None, *args, **kwa Returns: """ - return self.convert(structure, state_attr) + return self.convert(structure, state_attr, *args, **kwargs) @register @@ -147,7 +147,7 @@ def __init__( super().__init__(**kwargs) - def convert(self, structure: StructureOrMolecule, state_attr=None, **kwargs) -> MaterialGraph: + def convert(self, structure: StructureOrMolecule, state_attr=None, *args, **kwargs) -> MaterialGraph: """ Convert the structure into graph Args: