Revert no_past_residuals test change

Author: skirsten

tests/test_scheduler_flax.py

@@ -873,16 +873,11 @@ def test_pow_of_3_inference_steps(self):
 
     def test_inference_plms_no_past_residuals(self):
         with self.assertRaises(ValueError):
-            sample, _ = self.dummy_sample
-            num_inference_steps = 10
-
             scheduler_class = self.scheduler_classes[0]
             scheduler_config = self.get_scheduler_config()
             scheduler = scheduler_class(**scheduler_config)
             state = scheduler.create_state()
 
-            state = scheduler.set_timesteps(state, num_inference_steps, shape=sample.shape)
-
             scheduler.step_plms(state, self.dummy_sample, 1, self.dummy_sample).prev_sample
 
     def test_full_loop_no_noise(self):