toolchemy 0.2.194__tar.gz → 0.2.195__tar.gz

{toolchemy-0.2.194 → toolchemy-0.2.195}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: toolchemy
-Version: 0.2.194
+Version: 0.2.195
 Summary: a set of auxiliary programming tools
 License-File: LICENSE
 Author: Cyprian Nosek

{toolchemy-0.2.194 → toolchemy-0.2.195}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "toolchemy"
-version = "0.2.194"
+version = "0.2.195"
 description = "a set of auxiliary programming tools"
 authors = ["Cyprian Nosek <cyprian.nosek@protonmail.com>"]
 

{toolchemy-0.2.194 → toolchemy-0.2.195}/toolchemy/ai/trackers/common.py

@@ -93,6 +93,10 @@ class ITracker(ABC):
     def get_data(self) -> dict:
         pass
 
+    @abstractmethod
+    def get_traces(self, filter_name: str | None = None):
+        pass
+
 
 class TrackerBase(ITracker, ABC):
     def __init__(self, experiment_name: str, with_artifact_logging: bool = True, disabled: bool = False):
@@ -132,6 +136,9 @@ class TrackerBase(ITracker, ABC):
             "tags": self._tags.copy(),
         }
 
+    def get_traces(self, filter_name: str | None = None):
+        raise NotImplementedError()
+
     def _store_param(self, name: str, value: Any):
         if self._disabled:
             raise RuntimeError(f"Disabled trackers cannot store params!")

{toolchemy-0.2.194 → toolchemy-0.2.195}/toolchemy/ai/trackers/mlflow_tracker.py

@@ -195,6 +195,12 @@ class MLFlowTracker(TrackerBase):
         self._store_tag(name, value)
         self._client.set_experiment_tag(self._experiment_id, name, value)
 
+    def get_traces(self, filter_name: str | None = None):
+        filter_string = None
+        if filter_name:
+            filter_string = f"trace.name = '{filter_name}'"
+        return self._client.search_traces(experiment_ids=[self.experiment_id], run_id=self.run_id, filter_string=filter_string)
+
     def _reset_run(self):
         self._active_run = None
         self._active_run_id = None