Fixed few mypy warnings, disabling mypy warnings

Author: vanous

pygdtf/__init__.py

@@ -120,6 +120,8 @@ def __exit__(self, exc_type, exc_val, exc_tb):
             self._package.close()
 
     def _read_xml(self):
+        assert self._gdtf is not None
+        assert self._root is not None
         self.data_version = self._gdtf.get("DataVersion", 1.2)
         self.name = self._root.get("Name")
         self.short_name = self._root.get("ShortName")
@@ -312,7 +314,7 @@ def _read_xml(self):
         else:
             self.ft_presets = []
 
-        self.protocols = []
+        self.protocols: List[BaseNode] = []
         protocols_collect = self._root.find("Protocols")
         if protocols_collect is not None:
             for i in protocols_collect.findall("FTRDM"):
@@ -1275,7 +1277,7 @@ def flattened(self):
 
     def by_breaks(self, as_dict=False):
         # this is to unflatten the list
-        grouped = {}
+        grouped: Dict[int, List] = {}
 
         for item in self:
             key = item.dmx_break

pygdtf/geometries.py

@@ -82,7 +82,7 @@ def iterate_geometries(collector):
                 if hasattr(g, "geometries"):
                     iterate_geometries(g)
 
-        matched = []
+        matched: List["Geometry"] = []
         iterate_geometries(self)
         if matched:
             return matched[0]

pygdtf/utils/__init__.py

@@ -345,7 +345,8 @@ def calculate_complexity(gdtf_profile: Optional["pygdtf.FixtureType"] = None):
             slot
             for wheel in gdtf_profile.wheels
             for slot in wheel.wheel_slots
-            if slot.media_file_name.name is not None
+            if slot.media_file_name is not None
+            and slot.media_file_name.name is not None
         ]
     )
     facets_count = len(

pyproject.toml

@@ -64,3 +64,6 @@ dev = [
 Repository = "https://github.com/open-stage/python-gdtf.git"
 Changelog = "https://github.com/open-stage/python-gdtf/blob/master/CHANGELOG.md"
 
+[tool.mypy]
+check_untyped_defs = false
+disable_error_code = ["annotation-unchecked"]