numerai-tools 0.5.0.dev12__tar.gz → 0.5.0.dev13__tar.gz

{numerai_tools-0.5.0.dev12 → numerai_tools-0.5.0.dev13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: numerai-tools
-Version: 0.5.0.dev12
+Version: 0.5.0.dev13
 Summary: A collection of open-source tools to help interact with Numerai, model data, and automate submissions.
 License: MIT
 Author: Numerai Engineering

{numerai_tools-0.5.0.dev12 → numerai_tools-0.5.0.dev13}/numerai_tools/submissions.py

@@ -115,14 +115,18 @@ def validate_values(submission: pd.DataFrame, prediction_col: str) -> None:
         submission -- pandas DataFrame of the submission
         prediction_col -- the string name of the prediction column returned by validate_headers
     """
+    preds = submission[prediction_col]
     assert (
-        submission[prediction_col].isna().sum() == 0
+        preds.isna().sum() == 0
     ), "invalid_submission_values: submission must not contain NaNs"
     assert (
-        submission[prediction_col].between(0, 1).all()
+        preds.between(0, 1).all()
+        or (np.isclose(preds.min(), 0).all() and preds.le(1).all())
+        or (preds.ge(0).all() and np.isclose(preds.max(), 1).all())
+        or (np.isclose(preds.min(), 0).all() and np.isclose(preds.max(), 1).all())
     ), "invalid_submission_values: values must be between 0 and 1 exclusive"
     assert not np.isclose(
-        0, submission[prediction_col].std()
+        0, preds.std()
     ), "invalid_submission_values: submission must have non-zero standard deviation"
 
 
@@ -320,10 +324,6 @@ def clean_submission(
         pd.Series - the cleaned, properly indexed submission
     """
     assert len(universe) > 0, "universe must not be empty"
-    if isinstance(universe, pd.DataFrame):
-        assert universe.isna().sum().sum() == 0, "universe must not contain NaNs"
-    else:
-        assert universe.isna().sum() == 0, "universe must not contain NaNs"
     assert len(submission) > 0, "predictions must not be empty"
 
     if dst_id_col is None:

{numerai_tools-0.5.0.dev12 → numerai_tools-0.5.0.dev13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "numerai-tools"
-version = "0.5.0.dev12"
+version = "0.5.0.dev13"
 description = "A collection of open-source tools to help interact with Numerai, model data, and automate submissions."
 authors = [
     {name = "Numerai Engineering",email = "engineering@numer.ai"}