CFExecutor.py

from typing import Any, Callable
import concurrent.futures

from prefect.engine.executors.base import Executor


class CFExecutor(Executor):
    def __init__(self):
        print("Robexecutor")
        self.pool =  concurrent.futures.ThreadPoolExecutor(3)
        super().__init__()

    def submit(
        self, fn: Callable, *args: Any, extra_context: dict = None, **kwargs: Any
    ) -> Any:
        return self.pool.submit(fn, *args, **kwargs)

    def wait(self, futures: Any) -> Any:
        rval = concurrent.futures.wait(futures.values())
        print(rval)
        return rval