Fix: Implement better typing

src/aleph_vrf/coordinator/executor_selection.py

@@ -133,12 +133,10 @@ class UsePredeterminedExecutors(ExecutorSelectionPolicy):
     Use a hardcoded list of executors.
     """
 
-    def __init__(self, executors: List[Union[Executor, AlephExecutor]]):
+    def __init__(self, executors: List[VRFExecutor]):
         self.executors = executors
 
-    async def select_executors(
-        self, nb_executors: int
-    ) -> List[Union[Executor, AlephExecutor]]:
+    async def select_executors(self, nb_executors: int) -> List[VRFExecutor]:
         """
         Returns nb_executors from the hardcoded list of executors.
         If nb_executors is lower than the total number of executors, this method

src/aleph_vrf/coordinator/main.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Optional, Union
+from typing import List, Optional, Union
 
 from pydantic import BaseModel
 
@@ -23,6 +23,7 @@
     APIResponse,
     ComputeResourceNode,
     PublishedVRFResponse,
+    VRFExecutor,
 )
 
 logger.debug("imports done")
@@ -87,7 +88,9 @@ async def receive_test_vrf(
             "https://CRN_URL"  # CRN main URL, like https://hetzner.staging.aleph.sh
         )
         executor_node = ComputeResourceNode(address=executor_url, hash="", score=0)
-        executors = [AlephExecutor(node=executor_node, vm_function=settings.FUNCTION)]
+        executors: List[VRFExecutor] = [
+            AlephExecutor(node=executor_node, vm_function=settings.FUNCTION)
+        ]
         executor_policy = UsePredeterminedExecutors(executors)
         response = await generate_vrf(
             account=account,