Moving store object out of history attributes

Author: shreyashankar

mltrace/entities/base_component.py

@@ -500,4 +500,5 @@ def __repr__(self):
         params = self.to_dictionary()
         del params["beforeTests"]
         del params["afterTests"]
+        del params["history"]
         return json.dumps(params)

mltrace/entities/history.py

@@ -1,5 +1,3 @@
-
-
 import copy
 from datetime import datetime
 
@@ -13,15 +11,16 @@ def __init__(
         componentName: str,
     ):
         self.component_name = componentName
-        self.store = Store(clientUtils.get_db_uri())
 
     def get_runs_by_time(
         self,
         start_time: datetime = datetime.min,
         end_time: datetime = datetime.max,
     ):
-        history_runs = self.store.get_history(
-            self.component_name, None, start_time, end_time)
+        store = Store(clientUtils.get_db_uri())
+        history_runs = store.get_history(
+            self.component_name, None, start_time, end_time
+        )
         history_runs = clientUtils.convertToClient(history_runs)
         return history_runs
 
@@ -30,16 +29,24 @@ def get_runs_by_index(
         front_idx: int,
         last_idx: int,
     ):
-        history_runs = self.store.get_component_runs_by_index(
-            self.component_name, front_idx, last_idx)
+        store = Store(clientUtils.get_db_uri())
+        history_runs = store.get_component_runs_by_index(
+            self.component_name, front_idx, last_idx
+        )
         history_runs = clientUtils.convertToClient(history_runs)
         return history_runs
 
     def __getitem__(self, index):
-        history_run = self.store.get_component_runs_by_index(
-            self.component_name, index, index + 1)
+        store = Store(clientUtils.get_db_uri())
+        history_run = store.get_component_runs_by_index(
+            self.component_name, index, index + 1
+        )
         history_run = clientUtils.convertToClient(history_run)
         return history_run
 
     def __len__(self):
-        return self.store.get_component_runs_count(self.component_name)
+        store = Store(clientUtils.get_db_uri())
+        return store.get_component_runs_count(self.component_name)
+
+    def __repr__(self) -> str:
+        return f"History({self.component_name})"