Skip to content

Commit 27047bf

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 9dbbc8d commit 27047bf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tests_pytorch/trainer/optimization/test_optimizers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -704,7 +704,7 @@ def configure_optimizers(self):
704704

705705
# Calculate the total number of steps (iterations) and expected scheduler calls
706706
total_steps = 7 * 3 # Total iterations (7 batches per epoch * 3 epochs)
707-
expected_steps = (total_steps-1) // 5 # Scheduler steps every 5 iterations
707+
expected_steps = (total_steps - 1) // 5 # Scheduler steps every 5 iterations
708708

709709
# Assert that the scheduler was called the expected number of times
710710
assert mocked_sched.call_count == expected_steps

0 commit comments

Comments
 (0)