numerapi 2.23.0.dev2__py3-none-any.whl → 2.23.1__py3-none-any.whl

numerapi/__init__.py

@@ -3,9 +3,9 @@
 from importlib.metadata import version, PackageNotFoundError
 
 try:
-    __version__ = version("package-name")
+    __version__ = version("numerapi")
 except PackageNotFoundError:
-    __version__ = 'unknown'
+    __version__ = "unknown"
 
 
 # pylint: disable=wrong-import-position

numerapi/base_api.py

@@ -620,6 +620,79 @@ class Api:
         round_num = data["number"]
         return round_num
 
+    def list_rounds(
+        self,
+        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:
+            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
+              }
+            }
+        """
+        arguments = {
+            "tournament": self.tournament_id,
+            "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.
 
@@ -1054,7 +1127,6 @@ class Api:
         version: str | None = None,
         day: int | None = None,
         resolved: bool | None = None,
-        tournament: int | None = None,
         last_n_rounds: int | None = None,
         distinct_on_round: bool | None = None,
     ) -> List[Dict]:
@@ -1116,7 +1188,7 @@ class Api:
             "version": version,
             "day": day,
             "resolved": resolved,
-            "tournament": self.tournament_id if tournament is None else tournament,
+            "tournament": self.tournament_id,
             "lastNRounds": last_n_rounds,
             "distinctOnRound": distinct_on_round,
         }

{numerapi-2.23.0.dev2.dist-info → numerapi-2.23.1.dist-info}/METADATA

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

numerapi-2.23.1.dist-info/RECORD

@@ -0,0 +1,14 @@
+numerapi/__init__.py,sha256=15Ba_Ma1HCBVmSpTBbI-tjNGbfJUo2jagm5r7o2amT8,445
+numerapi/base_api.py,sha256=StASiaw3rX6RzV4uAkM3QEsPxF24LupwWgmKde9IIJ4,73357
+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.1.dist-info/licenses/LICENSE,sha256=BnOIJOdxTIxmbNPf_ZmCXqepXDTRWad1nQf9eYk-eSg,1056
+numerapi-2.23.1.dist-info/METADATA,sha256=gNRc6nbGSj1vmPrKOGH7KSHVVwXHdPHjhccMHKrGk-U,7036
+numerapi-2.23.1.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+numerapi-2.23.1.dist-info/entry_points.txt,sha256=P7RHLytfftNPE14vRxml52-UEkyuAviRegWTID4a_ig,46
+numerapi-2.23.1.dist-info/top_level.txt,sha256=7f4lKNQqRDEGaDXGIqRQCx-rU5pNl6ZgbXz864F1uXY,9
+numerapi-2.23.1.dist-info/RECORD,,

numerapi-2.23.0.dev2.dist-info/RECORD

@@ -1,14 +0,0 @@
-numerapi/__init__.py,sha256=okDA4NGjUj1Q8bDJg4T1t0LTf_S3GPbXCdw8gla7_VU,449
-numerapi/base_api.py,sha256=Q--49pamMwgb7AM98JC3gVS4K7k2xE7lmf6mtIwk3lc,71010
-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.dev2.dist-info/licenses/LICENSE,sha256=BnOIJOdxTIxmbNPf_ZmCXqepXDTRWad1nQf9eYk-eSg,1056
-numerapi-2.23.0.dev2.dist-info/METADATA,sha256=RtntW1DNgRIHzPTY19VrpNWOoo-XJmkm4jHBlhmKx74,7041
-numerapi-2.23.0.dev2.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-numerapi-2.23.0.dev2.dist-info/entry_points.txt,sha256=P7RHLytfftNPE14vRxml52-UEkyuAviRegWTID4a_ig,46
-numerapi-2.23.0.dev2.dist-info/top_level.txt,sha256=7f4lKNQqRDEGaDXGIqRQCx-rU5pNl6ZgbXz864F1uXY,9
-numerapi-2.23.0.dev2.dist-info/RECORD,,