Fix a few more mypy errors

DiamondLightSource · Sep 5, 2023 · 745c877 · 745c877
1 parent ee8fe89
commit 745c877
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/src/zocalo/configuration/__init__.py b/src/zocalo/configuration/__init__.py
@@ -311,7 +311,7 @@ def _read_configuration_yaml(configuration: str) -> dict:
                 f"Invalid YAML configuration: circular environment definitions for {environment_aliases}"
             )
 
-    plugin_fields = {}
+    plugin_fields: dict[str, mm.fields.Field] = {}
     for key in yaml_dict:
         if key in ConfigSchema().fields:
             continue

diff --git a/src/zocalo/service/__init__.py b/src/zocalo/service/__init__.py
@@ -29,7 +29,9 @@ def setup_logging(self):
 
         # Always enable logging to console
         try:
-            from dlstbx.util.colorstreamhandler import ColorStreamHandler
+            from dlstbx.util.colorstreamhandler import (
+                ColorStreamHandler,  # type: ignore
+            )
 
             self.console = ColorStreamHandler()
         except ImportError:

diff --git a/src/zocalo/wrapper.py b/src/zocalo/wrapper.py
@@ -4,6 +4,7 @@
 import threading
 from typing import Any, Callable
 
+import workflows.services.common_service
 import workflows.util
 
 import zocalo