test patched

pni-lab · Aug 8, 2021 · da4952f · da4952f
1 parent 15c57c1
commit da4952f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mlconfound/stats.py b/mlconfound/stats.py
@@ -291,8 +291,8 @@ def full_confound_test(y, yhat, c, num_perms=1000, cat_y=False, cat_yhat=False,
        --------
        See `notebooks/quickstart.ipynb` for more detailed examples.
 
-       >>> full_confound_test(y=[1,2,3,4,5,6], yhat=[1.5,2.3,2.9,4.2,5,5.7], c=[3,5,4,6,1,2], random_state=42).p
-       0.004
+       >>> full_confound_test(y=[1,2,3,4,5,6], yhat=[1.5,2.3,2.9,4.2,5,5.7], c=[3,5,4,6,1,2], random_state=42, num_perms=100).p
+       0.02
 
        See Also
        --------
@@ -398,8 +398,8 @@ def partial_confound_test(y, yhat, c, num_perms=1000, cat_y=False, cat_yhat=Fals
     --------
     See `notebooks/quickstart.ipynb` for more detailed examples.
 
-    >>> partial_confound_test(y=[1,2,3,4,5,6], yhat=[1.5,2.3,2.9,4.2,5,5.7], c=[3,5,4,6,1,2], random_state=42).p
-    0.692
+    >>> partial_confound_test(y=[1,2,3,4,5,6], yhat=[1.5,2.3,2.9,4.2,5,5.7], c=[3,5,4,6,1,2], random_state=42, num_perms=100).p
+    0.7
 
     See Also
     --------