added the changes

Author: 01AbhiSingh

tests/tests_pytorch/trainer/optimization/test_optimizers.py

@@ -703,7 +703,7 @@ def configure_optimizers(self):
 
     # Calculate the total number of steps (iterations) and expected scheduler calls
     total_steps = 7 * 3  # Total iterations (7 batches per epoch * 3 epochs)
-    expected_steps = total_steps // 5  # Scheduler steps every 5 iterations
+    expected_steps = (total_steps-1) // 5  # Scheduler steps every 5 iterations
 
     # Assert that the scheduler was called the expected number of times
     assert mocked_sched.call_count == expected_steps