File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed
tests/tests_pytorch/trainer/logging_ Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change @@ -745,17 +745,13 @@ def test_result_collection_metrics_include_pbar_parameter():
745
745
)
746
746
747
747
# 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 )
751
749
assert "pbar_metric" in metrics_with_pbar ["pbar" ]
752
750
assert "both_metric" in metrics_with_pbar ["pbar" ]
753
751
assert "both_metric" in metrics_with_pbar ["log" ]
754
752
755
753
# 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 )
759
755
# No progress bar metrics should be included
760
756
assert len (metrics_without_pbar ["pbar" ]) == 0
761
757
# Logger metrics should still be included
You can’t perform that action at this time.
0 commit comments