Mypy.

andrewtarzia · Sep 3, 2024 · 7843a67 · 7843a67
1 parent b45618a
commit 7843a67
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/cgexplore/_internal/utilities/chemiscope_interface.py b/src/cgexplore/_internal/utilities/chemiscope_interface.py
@@ -57,7 +57,7 @@ def __post_init__(self) -> None:
                 ],
             },
         }
-        self.all_shapes = {}
+        self.all_shapes: dict[str, dict] = {}
 
     def append_property(self, property_dict: dict) -> None:
         """Append properties in property_dict to json data."""
@@ -162,7 +162,7 @@ def append_bonds_as_shapes(self, molecule: stk.Molecule) -> None:
 
             shape_string = ",".join(self.all_shapes.keys())
 
-            self.json_data["settings"]["structure"][0]["shape"] = shape_string
+            self.json_data["settings"]["structure"][0]["shape"] = shape_string  # type: ignore[index]
 
     def write_json(self) -> None:
         """Write the chemiscope json."""