fix bug from new annotation style

CybercentreCanada · Sep 14, 2021 · 297f13e · 297f13e
1 parent e7ae18e
commit 297f13e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/assemblyline_core/server_base.py b/assemblyline_core/server_base.py
@@ -12,7 +12,7 @@
 import sys
 import io
 import os
-from typing import cast, Callable, TYPE_CHECKING
+from typing import Callable, TYPE_CHECKING
 
 from assemblyline.remote.datatypes import get_client
 from assemblyline.remote.datatypes.hash import Hash
@@ -208,7 +208,7 @@ def __init__(self, component_name: str, logger: logging.Logger = None,
         )
 
         # Create a cached service data object, and access to the service status
-        self.service_info = cast(dict[str, Service], forge.CachedObject(self._get_services))
+        self.service_info: dict[str, Service] = forge.CachedObject(self._get_services)
         self._service_stage_hash = get_service_stage_hash(self.redis)
 
     def _get_services(self):