update implementation to be more robust and documentation

AutoResearch · Jun 5, 2024 · df317af · df317af
1 parent 7f25d9b
commit df317af
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/autora/experimentalist/prediction_filter/__init__.py b/src/autora/experimentalist/prediction_filter/__init__.py
@@ -9,9 +9,9 @@
 
 
 def filter(
-        conditions:  Union[pd.DataFrame, np.ndarray],
-        model: BaseEstimator,
-        filter_function: Callable,
+    conditions: Union[pd.DataFrame, np.ndarray],
+    model: BaseEstimator,
+    filter_function: Callable,
 ) -> pd.DataFrame:
     """
     Filter conditions based on the expected outcome io the mdeol