test fix

open-thought · Mar 5, 2025 · c176b83 · c176b83
1 parent 72a74f3
commit c176b83
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_codeio.py b/tests/test_codeio.py
@@ -33,10 +33,10 @@ def test_codeio_dataset():
 def test_codeio_config():
     # Test constraints on input probability
     with pytest.raises(AssertionError):
-        CodeIOConfig(size=10, seed=42, input_prediction_probability=1.1)
+        CodeIOConfig(size=10, seed=42, input_prediction_probability=1.1).validate()
 
     with pytest.raises(AssertionError):
-        CodeIOConfig(size=10, seed=42, input_prediction_probability=-0.1)
+        CodeIOConfig(size=10, seed=42, input_prediction_probability=-0.1).validate()
 
-    CodeIOConfig(size=10, seed=42, input_prediction_probability=0.1)
-    CodeIOConfig(size=10, seed=42, input_prediction_probability=0.9)
+    CodeIOConfig(size=10, seed=42, input_prediction_probability=0.1).validate()
+    CodeIOConfig(size=10, seed=42, input_prediction_probability=0.9).validate()