format

kubeflow · Feb 12, 2025 · ed698f1 · ed698f1
1 parent e6b5888
commit ed698f1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pkg/metricscollector/v1beta1/tfevent-metricscollector/tfevent_loader.py b/pkg/metricscollector/v1beta1/tfevent-metricscollector/tfevent_loader.py
@@ -42,8 +42,9 @@ def _should_consider(tag: str, metric_name: str, tfefile: str) -> bool:
         else os.path.dirname(tfefile)
     )
     basedir_name = os.path.dirname(tfefile)
-    return (tag.startswith(metric_name.split("/")[-1]) and
-            basedir_name.endswith(tfefile_parent_dir))
+    return tag.startswith(metric_name.split("/")[-1]) and basedir_name.endswith(
+        tfefile_parent_dir
+    )
 
 
 class TFEventFileParser:

diff --git a/test/unit/v1beta1/metricscollector/test_tfevent_metricscollector.py b/test/unit/v1beta1/metricscollector/test_tfevent_metricscollector.py
@@ -17,10 +17,8 @@
 import unittest
 
 import tensorboardX
-
 import utils
 
-
 METRIC_DIR_NAMES = ("train", "test")
 METRIC_NAMES = ("accuracy", "loss")
 QUALIFIED_METRIC_NAMES = tuple(