fix: convert operation params to dict, preserve kwargs in industrial …

…decorator signature
aimclub · Jul 17, 2024 · ec7a3de · ec7a3de
1 parent d094221
commit ec7a3de
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/fedot_ind/core/architecture/abstraction/client.py b/fedot_ind/core/architecture/abstraction/client.py
@@ -13,9 +13,9 @@ def decorated_func(self, *args, **kwargs):
 
 
 def use_industrial_fedot_client(func):
-    def decorated_func(self, *args):
+    def decorated_func(self, *args, **kwargs):
         repo = IndustrialModels()
-        result = func(self, *args)
+        result = func(self, *args, **kwargs)
         repo.setup_repository()
         return result
 

diff --git a/fedot_ind/core/models/automl/fedot_implementation.py b/fedot_ind/core/models/automl/fedot_implementation.py
@@ -20,7 +20,7 @@ def __init__(self, params: Optional[OperationParameters] = None):
         super().__init__(params)
         if 'available_operations' not in self.params.keys():
             self.params.update({'available_operations': self.AVAILABLE_OPERATIONS})
-        self.model = Fedot(**self.params)
+        self.model = Fedot(**self.params.to_dict())
 
     def fit(self, input_data: InputData):
         self.model.fit(input_data)