Browse Source

Fix code regression in JobBase class.

Signed-off-by: Vittorio Cozzolino <vittorio.cozzolino@huawei.com>
tags/v0.5.1
Vittorio Cozzolino 3 years ago
parent
commit
92f83fc3fc
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      lib/sedna/core/base.py

+ 3
- 1
lib/sedna/core/base.py View File

@@ -106,6 +106,8 @@ class ModelLoadingThread(threading.Thread):

class JobBase:
""" sedna feature base class """
parameters = Context

def __init__(self, estimator, config=None):
self.config = BaseConfig()
if config:
@@ -158,7 +160,7 @@ class JobBase:
return callback_func(res) if callback_func else res

def get_parameters(self, param, default=None):
return Context.get_parameters(param=param, default=default)
return self.parameters.get_parameters(param=param, default=default)

def report_task_info(self, task_info, status, results=None, kind="train"):
message = {


Loading…
Cancel
Save