numerapi 2.23.0.dev1__py3-none-any.whl → 2.23.0.dev3__py3-none-any.whl

numerapi/base_api.py

@@ -14,6 +14,7 @@ import requests
 from numerapi import utils
 
 API_TOURNAMENT_URL = "https://api-tournament.numer.ai"
+_DEFAULT_TOURNAMENT = object()
 
 
 class Api:
@@ -427,7 +428,7 @@ class Api:
 
         max_amount = max_amount if max_amount is not None else amount
         query = """
-          mutation($submissionId: ID!, $staker: String!, $maxAmount: String!) {
+          query($submissionId: ID!, $staker: String!, $maxAmount: String!) {
             v3StakeAuth(
               submissionId: $submissionId
               staker: $staker
@@ -620,6 +621,84 @@ class Api:
         round_num = data["number"]
         return round_num
 
+    def list_rounds(
+        self,
+        tournament: int | None | object = _DEFAULT_TOURNAMENT,
+        number: int | None = None,
+        target: str | None = None,
+        status: str | None = None,
+        limit: int | None = None,
+    ) -> List[Dict]:
+        """List rounds with the filters supported by the round resolver.
+
+        Args:
+            tournament (int, optional): tournament filter, defaults to the API
+                instance tournament. Pass `None` to omit the tournament filter
+            number (int, optional): round number filter
+            target (str, optional): round target filter
+            status (str, optional): round status filter. One of `upcoming`,
+                `open`, `resolving`, or `resolved`
+            limit (int, optional): maximum number of rounds to return
+
+        Returns:
+            list of dicts: round entries matching the provided filters
+        """
+        query = """
+            query($tournament: Int
+                  $number: Int
+                  $target: String
+                  $status: RoundStatus
+                  $limit: Int) {
+              rounds(tournament: $tournament
+                     number: $number
+                     target: $target
+                     status: $status
+                     limit: $limit) {
+                id
+                tournament
+                number
+                target
+                closeTime
+                closeStakingTime
+                openTime
+                scoreTime
+                resolveTime
+                resolvedGeneral
+                resolvedStaking
+                payoutFactor
+                stakeThreshold
+                minCorrMultiplier
+                maxCorrMultiplier
+                defaultCorrMultiplier
+                minMmcMultiplier
+                maxMmcMultiplier
+                defaultMmcMultiplier
+                dataDatestamp
+              }
+            }
+        """
+        if tournament is _DEFAULT_TOURNAMENT:
+            tournament = self.tournament_id if self.tournament_id else None
+        arguments = {
+            "tournament": tournament,
+            "number": number,
+            "target": target,
+            "status": None if status is None else status.upper(),
+            "limit": limit,
+        }
+        rounds = self.raw_query(query, arguments)["data"]["rounds"]
+        for round_info in rounds:
+            for field in [
+                "closeTime",
+                "closeStakingTime",
+                "openTime",
+                "scoreTime",
+                "resolveTime",
+            ]:
+                utils.replace(round_info, field, utils.parse_datetime_string)
+            utils.replace(round_info, "payoutFactor", utils.parse_float_string)
+        return rounds
+
     def set_bio(self, model_id: str, bio: str) -> bool:
         """Set bio field for a model id.
 

{numerapi-2.23.0.dev1.dist-info → numerapi-2.23.0.dev3.dist-info}/METADATA

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

{numerapi-2.23.0.dev1.dist-info → numerapi-2.23.0.dev3.dist-info}/RECORD

@@ -1,14 +1,14 @@
 numerapi/__init__.py,sha256=okDA4NGjUj1Q8bDJg4T1t0LTf_S3GPbXCdw8gla7_VU,449
-numerapi/base_api.py,sha256=tdVIiy27RvKTpNTVZjdXVuCRjzzoOas_LO9N-MbGugo,71013
+numerapi/base_api.py,sha256=lAz19NA9WCF5VSo2V3IJXgqoLmXGXV9_rXzsGQTA1Qg,73800
 numerapi/cli.py,sha256=oaATypyxS0mlW2Uouby6Srq0DxWkouBg63uiXJY-QHM,8206
 numerapi/cryptoapi.py,sha256=J9fAEhiaEfpvRCiDsSJz8rlazKD-4nkJRJ85f09jGp0,1502
 numerapi/numerapi.py,sha256=cyTpNGRKr4BqK2jKA2Rj540tjyykpV3axz-6jLm1Flo,12323
 numerapi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 numerapi/signalsapi.py,sha256=inKwMvuIWNGcDFzzdhqRO1RIms_qz1d7lkwA2dzvAoQ,10188
 numerapi/utils.py,sha256=YgXujHyE1TLTf5v1pspcVAX89SQI3Zl3vMcbNlHZJDs,4688
-numerapi-2.23.0.dev1.dist-info/licenses/LICENSE,sha256=BnOIJOdxTIxmbNPf_ZmCXqepXDTRWad1nQf9eYk-eSg,1056
-numerapi-2.23.0.dev1.dist-info/METADATA,sha256=bgSq5xME7TXw3CnULv6QyN-3xkc5cBye5s39bcdym18,7041
-numerapi-2.23.0.dev1.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-numerapi-2.23.0.dev1.dist-info/entry_points.txt,sha256=P7RHLytfftNPE14vRxml52-UEkyuAviRegWTID4a_ig,46
-numerapi-2.23.0.dev1.dist-info/top_level.txt,sha256=7f4lKNQqRDEGaDXGIqRQCx-rU5pNl6ZgbXz864F1uXY,9
-numerapi-2.23.0.dev1.dist-info/RECORD,,
+numerapi-2.23.0.dev3.dist-info/licenses/LICENSE,sha256=BnOIJOdxTIxmbNPf_ZmCXqepXDTRWad1nQf9eYk-eSg,1056
+numerapi-2.23.0.dev3.dist-info/METADATA,sha256=FYh6HA-M0CsNHqwOQaQZzEOGQy1mMGCmBJgv4F6w1ng,7041
+numerapi-2.23.0.dev3.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+numerapi-2.23.0.dev3.dist-info/entry_points.txt,sha256=P7RHLytfftNPE14vRxml52-UEkyuAviRegWTID4a_ig,46
+numerapi-2.23.0.dev3.dist-info/top_level.txt,sha256=7f4lKNQqRDEGaDXGIqRQCx-rU5pNl6ZgbXz864F1uXY,9
+numerapi-2.23.0.dev3.dist-info/RECORD,,