chore: type fix 4

Exabyte-io · Jan 1, 2025 · c10aa41 · c10aa41
1 parent 2fd7c15
commit c10aa41
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/py/mat3ra/made/tools/build/defect/builders.py b/src/py/mat3ra/made/tools/build/defect/builders.py
@@ -127,9 +127,10 @@ class VoronoiInterstitialPointDefectBuilder(PointDefectBuilder):
         VoronoiInterstitialPointDefectBuilderParameters
     ) = VoronoiInterstitialPointDefectBuilderParameters  # type: ignore
     _DefaultBuildParameters = VoronoiInterstitialPointDefectBuilderParameters()  # type: ignore
-    _GeneratedItemType = PymatgenStructure
 
-    def _generate(self, configuration: BaseBuilder._ConfigurationType) -> List[_GeneratedItemType]:  # type: ignore
+    def _generate(
+        self, configuration: BaseBuilder._ConfigurationType
+    ) -> List[type(PointDefectBuilder._GeneratedItemType)]:  # type: ignore
         pymatgen_structure = to_pymatgen(configuration.crystal)
         voronoi_gen = PymatgenVoronoiInterstitialGenerator(
             **self.build_parameters.dict(),