FIX: Distributed Dataset

interTwin-eu · Aug 9, 2023 · 332b58d · 332b58d
1 parent ad30fa0
commit 332b58d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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):