Remove redundant training arg from tests

stefanradev93 · stefanradev93 · commit cfacbd8e666a · 2025-05-24T10:42:53.000-04:00
diff --git a/tests/test_networks/test_diffusion_model/test_diffusion_model.py b/tests/test_networks/test_diffusion_model/test_diffusion_model.py
@@ -17,7 +17,7 @@ def test_serialize_deserialize_noise_schedule(noise_schedule):
     assert noise_schedule.get_drift_diffusion(t, x, training=False) == deserialized.get_drift_diffusion(
         t, x, training=False
     )
-    assert noise_schedule.get_alpha_sigma(t, training=training) == deserialized.get_alpha_sigma(t, training=training)
+    assert noise_schedule.get_alpha_sigma(t) == deserialized.get_alpha_sigma(t)
     assert noise_schedule.get_weights_for_snr(t) == deserialized.get_weights_for_snr(t)
 
 

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ def test_serialize_deserialize_noise_schedule(noise_schedule):`
`17`	`17`	`assert noise_schedule.get_drift_diffusion(t, x, training=False) == deserialized.get_drift_diffusion(`
`18`	`18`	`t, x, training=False`
`19`	`19`	`)`
`20`		`- assert noise_schedule.get_alpha_sigma(t, training=training) == deserialized.get_alpha_sigma(t, training=training)`
	`20`	`+ assert noise_schedule.get_alpha_sigma(t) == deserialized.get_alpha_sigma(t)`
`21`	`21`	`assert noise_schedule.get_weights_for_snr(t) == deserialized.get_weights_for_snr(t)`
`22`	`22`
`23`	`23`