From 332b58dbc3b273ec0872aa2beaed231556578396 Mon Sep 17 00:00:00 2001 From: User3574 Date: Wed, 9 Aug 2023 15:19:19 -0400 Subject: [PATCH] FIX: Distributed Dataset --- ai/src/itwinai/backend/executors.py | 4 ++++ 1 file changed, 4 insertions(+) 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):