numerapi 2.20.6__tar.gz → 2.20.8__tar.gz

{numerapi-2.20.6 → numerapi-2.20.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: numerapi
-Version: 2.20.6
+Version: 2.20.8
 Summary: Automatically download and upload data for the Numerai machine learning competition
 Home-page: https://github.com/uuazed/numerapi
 Maintainer: uuazed

{numerapi-2.20.6 → numerapi-2.20.8}/numerapi/base_api.py

@@ -60,7 +60,8 @@ class Api:
                 "You need to supply both a public id and a secret key.")
             self.token = None
 
-    def _handle_call_error(self, errors):
+    def _handle_call_error(self, errors) -> str:
+        msg = ""
         if isinstance(errors, list):
             for error in errors:
                 if "message" in error:
@@ -157,7 +158,7 @@ class Api:
         args = {'round': round_num, "tournament": self.tournament_id}
         return self.raw_query(query, args)['data']['listDatasets']
 
-    def download_dataset(self, filename: str = None,
+    def download_dataset(self, filename: str,
                          dest_path: str | None = None,
                          round_num: int | None = None) -> str:
         """ Download specified file for the given round.
@@ -189,12 +190,14 @@ class Api:
 
         query = """
         query ($filename: String!
-               $round: Int) {
+               $round: Int
+               $tournament: Int) {
             dataset(filename: $filename
-                    round: $round)
+                    round: $round
+                    tournament: $tournament)
         }
         """
-        args = {'filename': filename, "round": round_num}
+        args = {'filename': filename, "round": round_num, "tournament": self.tournament_id}
 
         dataset_url = self.raw_query(query, args)['data']['dataset']
         utils.download_file(dataset_url, dest_path, self.show_progress_bars)
@@ -323,7 +326,7 @@ class Api:
         return {item["displayName"]: item["id"]
                 for item in sorted(data, key=lambda x: x["displayName"])}
 
-    def get_models(self, tournament: int = None) -> Dict:
+    def get_models(self, tournament: int | None = None) -> Dict:
         """Get mapping of account model names to model ids for convenience
 
         Args:
@@ -500,7 +503,7 @@ class Api:
         return txs
 
     def set_submission_webhook(self, model_id: str,
-                               webhook: str = None) -> bool:
+                               webhook: str | None = None) -> bool:
         """Set a model's submission webhook used in Numerai Compute.
         Read More: https://docs.numer.ai/tournament/compute
 
@@ -555,7 +558,7 @@ class Api:
 
     def upload_diagnostics(self, file_path: str = "predictions.csv",
                            tournament: int | None = None,
-                           model_id: str | None = None,
+                           model_id: str = "",
                            df: pd.DataFrame | None = None) -> str:
         """Upload predictions to diagnostics from file.
 
@@ -634,6 +637,8 @@ class Api:
                 * perEraDiagnostics (`list`) each with the following fields:
                     * era (`int`)
                     * examplePredsCorr (`float`)
+                    * validationAlpha (`float`)
+                    * validationBmc (`float`)
                     * validationCorr (`float`)
                     * validationCorrV4 (`float`)
                     * validationFeatureCorrMax (`float`)
@@ -695,6 +700,11 @@ class Api:
                 * validationRicMean (`float`)
                 * validationRicSharpe (`float`)
                 * validationRicStd (`float`)
+                * validationAlphaCorrWExamplePreds (`float`)
+                * validationAlphaMaxDrawdown (`float`)
+                * validationAlphaMean (`float`)
+                * validationAlphaSharpe (`float`)
+                * validationAlphaStd (`float`)
 
         Example:
             >>> napi = NumerAPI(secret_key="..", public_id="..")
@@ -718,6 +728,9 @@ class Api:
                 perEraDiagnostics {
                     era
                     examplePredsCorr
+                    validationAlpha
+                    validationBmc
+                    validationChurn
                     validationCorr
                     validationCorrV4
                     validationFeatureCorrMax
@@ -727,7 +740,6 @@ class Api:
                     validationFncV4
                     validationIcV2
                     validationRic
-                    validationBmc
                 }
                 status
                 trainedOnVal
@@ -787,6 +799,11 @@ class Api:
                 validationRicMean
                 validationRicSharpe
                 validationRicStd
+                validationAlphaCorrWExamplePreds
+                validationAlphaMaxDrawdown
+                validationAlphaMean
+                validationAlphaSharpe
+                validationAlphaStd
               }
             }
         '''
@@ -935,7 +952,7 @@ class Api:
 
 
     def stake_change(self, nmr, action: str = "decrease",
-                     model_id: str | None = None) -> Dict:
+                     model_id: str = "") -> Dict:
         """Change stake by `value` NMR.
 
         Args:
@@ -1031,7 +1048,7 @@ class Api:
         raw = self.raw_query(query, arguments, authorization=True)
         return raw['data']['releaseStake']
 
-    def stake_decrease(self, nmr, model_id: str | None = None) -> Dict:
+    def stake_decrease(self, nmr, model_id: str) -> Dict:
         """Decrease your stake by `value` NMR.
 
         Args:
@@ -1059,7 +1076,7 @@ class Api:
         """
         return self.stake_change(nmr, 'decrease', model_id)
 
-    def stake_increase(self, nmr, model_id: str | None = None) -> Dict:
+    def stake_increase(self, nmr, model_id: str) -> Dict:
         """Increase your stake by `value` NMR.
 
         Args:
@@ -1470,8 +1487,8 @@ class Api:
         return data
 
     def download_submission(self, submission_id: str | None = None,
-                            model_id: str | None = None,
-                            dest_path: str | None = None) -> str:
+                            model_id: str = "",
+                            dest_path: str = "") -> str:
         """ Download previous submissions from numerai
 
         Args:
@@ -1494,7 +1511,7 @@ class Api:
             >>> api.download_submission(submission_id=submission_id)
         """
         msg = "You need to provide one of `model_id` and `submission_id"
-        assert model_id or submission_id, msg
+        assert model_id != "" or submission_id != "", msg
         auth_query = '''
             query($id: String) {
                 submissionDownloadAuth(id: $id) {
@@ -1509,7 +1526,7 @@ class Api:
         data = self.raw_query(
             auth_query, {'id': submission_id},
             authorization=True)['data']["submissionDownloadAuth"]
-        if dest_path is None:
+        if dest_path == "":
             dest_path = data["filename"]
         path = utils.download_file(data["url"], dest_path)
         return path

{numerapi-2.20.6 → numerapi-2.20.8}/numerapi/utils.py

@@ -104,7 +104,7 @@ def download_file(url: str, dest_path: str, show_progress_bars: bool = True):
     return dest_path
 
 
-def post_with_err_handling(url: str, body: str, headers: dict,
+def post_with_err_handling(url: str, body: dict, headers: dict,
                            *, timeout: int | None = None,
                            retries: int = 3, delay: int = 1, backoff: int = 2
                            ) -> dict:

{numerapi-2.20.6 → numerapi-2.20.8}/numerapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: numerapi
-Version: 2.20.6
+Version: 2.20.8
 Summary: Automatically download and upload data for the Numerai machine learning competition
 Home-page: https://github.com/uuazed/numerapi
 Maintainer: uuazed

{numerapi-2.20.6 → numerapi-2.20.8}/setup.py

@@ -6,7 +6,7 @@ def load(path):
     return open(path, 'r').read()
 
 
-numerapi_version = '2.20.6'
+numerapi_version = '2.20.8'
 
 
 classifiers = [