Skip to content

Commit bfdfad3

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 293233c commit bfdfad3

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tests/tests_pytorch/trainer/logging_/test_logger_connector.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -745,17 +745,13 @@ def test_result_collection_metrics_include_pbar_parameter():
745745
)
746746

747747
# Test with include_pbar_metrics=True (default behavior)
748-
metrics_with_pbar = results.metrics(
749-
on_step=True, include_pbar_metrics=True
750-
)
748+
metrics_with_pbar = results.metrics(on_step=True, include_pbar_metrics=True)
751749
assert "pbar_metric" in metrics_with_pbar["pbar"]
752750
assert "both_metric" in metrics_with_pbar["pbar"]
753751
assert "both_metric" in metrics_with_pbar["log"]
754752

755753
# Test with include_pbar_metrics=False (optimization)
756-
metrics_without_pbar = results.metrics(
757-
on_step=True, include_pbar_metrics=False
758-
)
754+
metrics_without_pbar = results.metrics(on_step=True, include_pbar_metrics=False)
759755
# No progress bar metrics should be included
760756
assert len(metrics_without_pbar["pbar"]) == 0
761757
# Logger metrics should still be included

0 commit comments

Comments
 (0)