diff --git a/src/cgexplore/_internal/utilities/chemiscope_interface.py b/src/cgexplore/_internal/utilities/chemiscope_interface.py index 904d1d8c..aacac410 100644 --- 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."""