diff --git a/mlflow_go/store/tracking.py b/mlflow_go/store/tracking.py index d8694b0..80767ef 100644 --- a/mlflow_go/store/tracking.py +++ b/mlflow_go/store/tracking.py @@ -19,7 +19,6 @@ GetRun, LogBatch, LogMetric, - LogParam, RestoreExperiment, RestoreRun, SearchRuns, @@ -166,6 +165,7 @@ def log_metric(self, run_id, metric): ) self.service.call_endpoint(get_lib().TrackingServiceLogMetric, request) + def TrackingStore(cls): return type(cls.__name__, (_TrackingStore, cls), {})