Skip to content

Commit ed698f1

Browse files
committed
format
1 parent e6b5888 commit ed698f1

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

pkg/metricscollector/v1beta1/tfevent-metricscollector/tfevent_loader.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,9 @@ def _should_consider(tag: str, metric_name: str, tfefile: str) -> bool:
4242
else os.path.dirname(tfefile)
4343
)
4444
basedir_name = os.path.dirname(tfefile)
45-
return (tag.startswith(metric_name.split("/")[-1]) and
46-
basedir_name.endswith(tfefile_parent_dir))
45+
return tag.startswith(metric_name.split("/")[-1]) and basedir_name.endswith(
46+
tfefile_parent_dir
47+
)
4748

4849

4950
class TFEventFileParser:

test/unit/v1beta1/metricscollector/test_tfevent_metricscollector.py

-2
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,8 @@
1717
import unittest
1818

1919
import tensorboardX
20-
2120
import utils
2221

23-
2422
METRIC_DIR_NAMES = ("train", "test")
2523
METRIC_NAMES = ("accuracy", "loss")
2624
QUALIFIED_METRIC_NAMES = tuple(

0 commit comments

Comments
 (0)