fix mypy issues

materialsproject · Jun 7, 2021 · 744fde9 · 744fde9
1 parent 2ddf348
commit 744fde9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/emmet-core/emmet/core/oxidation_states.py b/emmet-core/emmet/core/oxidation_states.py
@@ -28,7 +28,9 @@ class OxidationStateDoc(PropertyDoc):
     method: str = Field(description="Method used to compute oxidation states")
 
     @classmethod
-    def from_structure(cls, structure: Structure, material_id: MPID):
+    def from_structure(
+        cls, structure: Structure, material_id: MPID, **kwargs  # type: ignore[override]
+    ):
         structure.remove_oxidation_states()
         try:
             bva = BVAnalyzer()
@@ -89,4 +91,6 @@ def from_structure(cls, structure: Structure, material_id: MPID):
                 logging.error("Oxidation state guess failed with: {}".format(e))
                 raise e
 
-        return super().from_structure(structure=structure, material_id=material_id, **d)
+        return super().from_structure(
+            structure=structure, material_id=material_id, **d, **kwargs
+        )