refactor: fixed connector cls type

bec_ipython_client/bec_ipython_client/main.py

@@ -41,8 +41,8 @@ class CLIBECClient(BECClient):
 class BECIPythonClient:
     def __init__(
         self,
-        config: ServiceConfig = None,
-        connector_cls: RedisConnector = None,
+        config: ServiceConfig | None = None,
+        connector_cls: type[RedisConnector] | None = None,
         wait_for_server=True,
         forced=False,
     ) -> None:

bec_lib/bec_lib/client.py

@@ -99,8 +99,8 @@ class BECClient(BECService, UserScriptsMixin):
 
     def __init__(
         self,
-        config: ServiceConfig = None,
-        connector_cls: RedisConnector = None,
+        config: ServiceConfig | None = None,
+        connector_cls: type[RedisConnector] | None = None,
         wait_for_server=False,
         forced=False,
         parent=None,

bec_server/bec_server/data_processing/dap_server.py

@@ -15,7 +15,7 @@ class DAPServer(BECClient):
     def __init__(
         self,
         config: ServiceConfig,
-        connector_cls: RedisConnector,
+        connector_cls: type[RedisConnector],
         provided_services: list,
         forced=True,
     ) -> None:

bec_server/bec_server/device_server/device_server.py

@@ -209,7 +209,7 @@ class DeviceServer(RPCMixin, BECService):
     This class is intended to provide a thin wrapper around ophyd and the devicemanager. It acts as the entry point for other services
     """
 
-    def __init__(self, config, connector_cls: RedisConnector) -> None:
+    def __init__(self, config, connector_cls: type[RedisConnector]) -> None:
         super().__init__(config, connector_cls, unique_service=True)
         self._tasks = []
         self.device_manager = None

bec_server/bec_server/file_writer/file_writer_manager.py

@@ -64,7 +64,7 @@ class ScanStorage:
 
 
 class FileWriterManager(BECService):
-    def __init__(self, config: ServiceConfig, connector_cls: RedisConnector) -> None:
+    def __init__(self, config: ServiceConfig, connector_cls: type[RedisConnector]) -> None:
         """
         Service to write scan data to file.
 

bec_server/bec_server/scan_bundler/scan_bundler.py

@@ -25,7 +25,7 @@ logger = bec_logger.logger
 
 
 class ScanBundler(BECService):
-    def __init__(self, config, connector_cls: RedisConnector) -> None:
+    def __init__(self, config, connector_cls: type[RedisConnector]) -> None:
         super().__init__(config, connector_cls, unique_service=True)
 
         self.device_manager = None

bec_server/bec_server/scan_server/scan_server.py

@@ -29,7 +29,7 @@ class ScanServer(BECService):
     scan_assembler = None
     scan_manager = None
 
-    def __init__(self, config: ServiceConfig, connector_cls: RedisConnector):
+    def __init__(self, config: ServiceConfig, connector_cls: type[RedisConnector]):
         super().__init__(config, connector_cls, unique_service=True)
         self._start_scan_manager()
         self._start_queue_manager()

bec_server/bec_server/scihub/scihub.py

@@ -14,7 +14,7 @@ if TYPE_CHECKING:
 
 
 class SciHub(BECService):
-    def __init__(self, config: ServiceConfig, connector_cls: RedisConnector) -> None:
+    def __init__(self, config: ServiceConfig, connector_cls: type[RedisConnector]) -> None:
         super().__init__(config, connector_cls, unique_service=True)
         self.config = config
         self.atlas_connector = None