numerai-tools 0.5.0.dev6__tar.gz → 0.5.0.dev7__tar.gz

{numerai_tools-0.5.0.dev6 → numerai_tools-0.5.0.dev7}/PKG-INFO

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

@@ -107,6 +107,7 @@ def calculate_max_churn_and_turnover(
     (
         curr_ticker_col,
         curr_signal_col,
+        _,
         curr_sub,
         _,
     ) = validate_submission_signals(
@@ -135,6 +136,7 @@ def calculate_max_churn_and_turnover(
         (
             prev_ticker_col,
             prev_signal_col,
+            _,
             prev_sub,
             _,
         ) = validate_submission_signals(

{numerai_tools-0.5.0.dev6 → numerai_tools-0.5.0.dev7}/numerai_tools/submissions.py

@@ -206,8 +206,8 @@ def validate_submission_numerai(
 
 
 def validate_submission_signals(
-    universe: pd.DataFrame, submission: pd.DataFrame
-) -> Tuple[str, str, pd.DataFrame, List[str]]:
+    universe: pd.DataFrame, submission: pd.DataFrame, assert_date_col: bool = False
+) -> Tuple[str, str, Optional[str], pd.DataFrame, List[str]]:
     """Validate the headers, ids, and values for a submission.
 
     Arguments:
@@ -224,13 +224,19 @@ def validate_submission_signals(
             "data_type column found in Signals submission. This is deprecated and support will be removed in the future. "
             "Please remove the data_type column from your Signals submission."
         )
-        submission = submission.drop(columns=["data_type"], errors="ignore")
-    ticker_col, signal_col, _ = validate_headers_signals(submission)
+        submission.drop(
+            columns=["data_type"],
+            errors="ignore",
+            inplace=True,
+        )
+    ticker_col, signal_col, date_col = validate_headers_signals(
+        submission, assert_date_col
+    )
     filtered_sub, invalid_tickers = validate_ids_signals(
         universe[ticker_col], submission, ticker_col
     )
     validate_values(filtered_sub, signal_col)
-    return ticker_col, signal_col, filtered_sub, invalid_tickers
+    return ticker_col, signal_col, date_col, filtered_sub, invalid_tickers
 
 
 def validate_submission_crypto(

{numerai_tools-0.5.0.dev6 → numerai_tools-0.5.0.dev7}/pyproject.toml

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