chore(tests/sampling): drop redundant print statement

ml-gde · Oct 9, 2024 · 464b094 · 464b094
1 parent 55f378b
commit 464b094
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/test_sampling.py b/tests/test_sampling.py
@@ -33,5 +33,4 @@ def test_get_noise(self):
             seed=42,
             device="cuda" if torch.cuda.is_available() else "cpu",
         )
-        print(x_jax.shape)
         chex.assert_equal_shape([x_jax, x_torch])