Update __init__.py

AutoResearch · Oct 14, 2024 · 1c9a85d · 1c9a85d
1 parent efdb238
commit 1c9a85d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/autora/experimentalist/falsification/__init__.py b/src/autora/experimentalist/falsification/__init__.py
@@ -24,7 +24,7 @@ def pool(
     reference_conditions: Union[pd.DataFrame, np.ndarray],
     reference_observations: Union[pd.DataFrame, np.ndarray],
     metadata: VariableCollection,
-    num_samples: int = 100,
+    num_samples: int = 1,
     training_epochs: int = 1000,
     optimization_epochs: int = 1000,
     training_lr: float = 1e-3,
@@ -197,7 +197,7 @@ def sample(
     reference_conditions: Union[pd.DataFrame, np.ndarray],
     reference_observations: Union[pd.DataFrame, np.ndarray],
     metadata: VariableCollection,
-    num_samples: Optional[int] = None,
+    num_samples: int = 1,
     training_epochs: int = 1000,
     training_lr: float = 1e-3,
     plot: bool = False,
@@ -283,7 +283,7 @@ def falsification_score_sample(
     reference_conditions: Union[pd.DataFrame, np.ndarray],
     reference_observations: Union[pd.DataFrame, np.ndarray],
     metadata: Optional[VariableCollection] = None,
-    num_samples: Optional[int] = None,
+    num_samples: int = 1,
     training_epochs: int = 1000,
     training_lr: float = 1e-3,
     plot: bool = False,
@@ -357,7 +357,7 @@ def falsification_score_sample_from_predictions(
     reference_conditions: Union[pd.DataFrame, np.ndarray],
     reference_observations: np.ndarray,
     metadata: Optional[VariableCollection] = None,
-    num_samples: Optional[int] = None,
+    num_samples: int = 1,
     training_epochs: int = 1000,
     training_lr: float = 1e-3,
     plot: bool = False,