change/rename imports

comet-ml · Feb 18, 2025 · 45ee2f3 · 45ee2f3
1 parent cbe1254
commit 45ee2f3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sdks/python/src/opik/evaluation/metrics/base_metric.py b/sdks/python/src/opik/evaluation/metrics/base_metric.py
@@ -1,7 +1,8 @@
 import abc
 from typing import Any, List, Union
 
-from opik import config as opik_config, track as track_decorator
+import opik
+from opik import config as opik_config
 from ..metrics import score_result
 
 
@@ -39,8 +40,8 @@ def __init__(self, name: str, track: bool = True) -> None:
         config = opik_config.OpikConfig()
 
         if track and opik_config.is_misconfigured(config) is False:
-            self.score = track_decorator(name=self.name)(self.score)  # type: ignore
-            self.ascore = track_decorator(name=self.name)(self.ascore)  # type: ignore
+            self.score = opik.track(name=self.name)(self.score)  # type: ignore
+            self.ascore = opik.track(name=self.name)(self.ascore)  # type: ignore
 
     @abc.abstractmethod
     def score(