toolchemy 0.2.193__tar.gz → 0.2.195__tar.gz

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

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

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

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

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

@@ -11,11 +11,21 @@ class ITracker(ABC):
     def experiment_name(self) -> str:
         pass
 
+    @property
+    @abstractmethod
+    def experiment_id(self) -> str:
+        pass
+
     @property
     @abstractmethod
     def run_name(self) -> str:
         pass
 
+    @property
+    @abstractmethod
+    def run_id(self) -> str:
+        pass
+
     @abstractmethod
     def start_run(
             self, run_id: str = None,
@@ -83,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):
@@ -101,6 +115,10 @@ class TrackerBase(ITracker, ABC):
     def experiment_name(self) -> str:
         return self._experiment_name
 
+    @property
+    def experiment_id(self) -> str:
+        return self._experiment_name
+
     def get_max_metric_value(self, name: str) -> float:
         return max(self._metrics[name], key=lambda el: el['value'])
 
@@ -118,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!")
@@ -166,6 +187,10 @@ class InMemoryTracker(TrackerBase):
     def run_name(self) -> str:
         return self._run_name
 
+    @property
+    def run_id(self) -> str:
+        return self._run_name
+
     def start_run(
             self, run_id: str = None,
             run_name: str = None,

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

@@ -39,6 +39,16 @@ class MLFlowTracker(TrackerBase):
             raise RuntimeError("There is no active run!")
         return self._active_run.info.run_name
 
+    @property
+    def run_id(self) -> str:
+        if not self._active_run:
+            raise RuntimeError("There is no active run!")
+        return self._active_run.info.run_id
+
+    @property
+    def experiment_id(self) -> str:
+        return self._experiment_id
+
     def start_run(
             self, run_id: str = None,
             run_name: str = None,
@@ -185,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

{toolchemy-0.2.193 → toolchemy-0.2.195}/toolchemy/ai/trackers/neptune_tracker.py

@@ -20,6 +20,10 @@ class NeptuneAITracker(TrackerBase):
             raise RuntimeError("There is no active run!")
         return self._active_run_id
 
+    @property
+    def run_id(self) -> str:
+        return self.run_name
+
 
     def start_run(
             self, run_id: str = None,