diff --git a/ai/src/itwinai/backend/executors.py b/ai/src/itwinai/backend/executors.py index 9acfa7f7..4a18e9a0 100644 --- a/ai/src/itwinai/backend/executors.py +++ b/ai/src/itwinai/backend/executors.py @@ -1,4 +1,5 @@ import yaml +import ray from .components import Executor from jsonargparse import ArgumentParser @@ -40,6 +41,9 @@ def __init__(self, pipeline, class_dict, param_space): self.class_dict = class_dict self.param_space = param_space + # Init ray + ray.init() + def worker_fn(self, config, pipeline, class_dict): # Should have same structure pipe and params def replace(pipe, params):