We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9dbbc8d commit 27047bfCopy full SHA for 27047bf
tests/tests_pytorch/trainer/optimization/test_optimizers.py
@@ -704,7 +704,7 @@ def configure_optimizers(self):
704
705
# Calculate the total number of steps (iterations) and expected scheduler calls
706
total_steps = 7 * 3 # Total iterations (7 batches per epoch * 3 epochs)
707
- expected_steps = (total_steps-1) // 5 # Scheduler steps every 5 iterations
+ expected_steps = (total_steps - 1) // 5 # Scheduler steps every 5 iterations
708
709
# Assert that the scheduler was called the expected number of times
710
assert mocked_sched.call_count == expected_steps
0 commit comments