From 4eb8338de5eac27ec2e504b0b46adf42302f9ea9 Mon Sep 17 00:00:00 2001
From: Tsz Wai Ko <47970742+kenko911@users.noreply.github.com>
Date: Thu, 26 Sep 2024 15:09:30 -0700
Subject: [PATCH] further fix Too many positional arguments for method call in
 _converters.py

---
 m3gnet/graph/_converters.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/m3gnet/graph/_converters.py b/m3gnet/graph/_converters.py
index ac5afee..06594c7 100644
--- a/m3gnet/graph/_converters.py
+++ b/m3gnet/graph/_converters.py
@@ -78,7 +78,7 @@ def get_states(self, structure: StructureOrMolecule):
         return states
 
     @abstractmethod
-    def convert(self, structure: StructureOrMolecule, **kwargs) -> MaterialGraph:
+    def convert(self, structure: StructureOrMolecule, state_attr=None, *args, **kwargs) -> MaterialGraph:
         """
         Convert the structure into a graph
         Args: