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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: numerai-tools
-Version: 0.5.0.dev11
+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.dev11 → numerai_tools-0.5.0.dev13}/numerai_tools/signals.py

@@ -9,7 +9,6 @@ from numerai_tools.scoring import (
 from numerai_tools.submissions import (
     validate_submission_signals,
     clean_submission,
-    remap_ids,
 )
 
 import pandas as pd
@@ -121,7 +120,6 @@ def calculate_max_churn_and_turnover(
         src_signal_col=curr_signal_col,
         rank_and_fill=True,
     )
-    print("curr_sub", curr_sub)
     churn_stats = []
     turnover_stats = []
     neutralized_weights = generate_neutralized_weights(
@@ -159,7 +157,6 @@ def calculate_max_churn_and_turnover(
             prev_sample_weight,
             center_and_normalize=True,
         )[prev_sub.name]
-        print("prev_sub", prev_sub)
         try:
             churn_val = abs(churn(curr_sub, prev_sub))
         except AssertionError as e:

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

@@ -88,7 +88,6 @@ def validate_headers_signals(
         assert (
             date_col_name is not None
         ), "invalid_submission_headers: submission must contain a date column"
-    print(submission)
     ticker_col, signal_col = _validate_headers(
         submission,
         SIGNALS_ALLOWED_ID_COLS,
@@ -116,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"
 
 
@@ -230,7 +233,6 @@ def validate_submission_signals(
             "Please remove the data_type column from your Signals submission."
         )
     submission.drop(columns=["data_type"], errors="ignore", inplace=True)
-    print(submission)
     ticker_col, signal_col, date_col = validate_headers_signals(
         submission, assert_date_col
     )
@@ -322,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.dev11 → numerai_tools-0.5.0.dev13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "numerai-tools"
-version = "0.5.0.dev11"
+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"}