codebase cleanup

Author: Boyuan Deng

examples/boyuan_component_get_history_test.py

@@ -4,18 +4,18 @@
 class BoyuanComponent(Component):
     def __init__(
         self,
-        name = "",
-        owner = "",
-        description = "",
-        beforeTests = [],
-        afterTests = [],
-        tags = [],
+        name="",
+        owner="",
+        description="",
+        beforeTests=[],
+        afterTests=[],
+        tags=[],
     ):
         super().__init__(name, owner, description, beforeTests, afterTests, tags)
 
     # overwrite afterRun
     def afterRun(self):
-        histories = self._history.get_runs_by_index(0,1)
+        histories = self._history.get_runs_by_index(0, 1)
         history_length = len(self._history)
 
 

mltrace/db/store.py

@@ -562,8 +562,7 @@ def get_component_runs_by_index(
         return history
 
     def get_component_runs_count(self, component_name: str):
-        return  self.session.query(ComponentRun).count()
-
+        return self.session.query(ComponentRun).count()
 
     def get_components(self, tag: str = "", owner: str = ""):
         """Returns a list of all the components associated with the specified