rustat-python-api 0.5.6__tar.gz → 0.5.8__tar.gz

{rustat-python-api-0.5.6/rustat_python_api.egg-info → rustat-python-api-0.5.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustat-python-api
-Version: 0.5.6
+Version: 0.5.8
 Summary: A Python wrapper for RuStat API
 Home-page: https://github.com/dailydaniel/rustat-python-api
 Author: Daniel Zholkovsky

{rustat-python-api-0.5.6 → rustat-python-api-0.5.8}/rustat_python_api/parser.py

@@ -32,6 +32,20 @@ class RuStatParser:
         response = requests.get(query)
         return response.json()
 
+    def get_season_teams(self, season_id: int):
+        data = self.resp2data(
+            self.urls["tournament_teams"].format(
+                user=self.user,
+                password=self.password,
+                season_id=season_id
+            )
+        )
+
+        if data:
+            return data['data']['row']
+        else:
+            return []
+
     def get_rpl_info(self):
         for season_id in tqdm(range(1, 36)):
             data = self.resp2data(

{rustat-python-api-0.5.6 → rustat-python-api-0.5.8}/rustat_python_api/pitch_control.py

@@ -148,7 +148,8 @@ class PitchControl:
         location: np.ndarray,
         time_index: int,
         home_or_away: str,
-        half: int
+        half: int,
+        verbose: bool = False
     ):
         if home_or_away == 'h':
             data = self.locs_home[half].copy()
@@ -193,11 +194,12 @@ class PitchControl:
             Sigma = np.matmul(Sigma, np.linalg.inv(R)) ## this is not efficient, forgive me.
             out = mvn.pdf(location, mu, Sigma) / mvn.pdf(data[player_index][time_index, :], mu, Sigma)
         else:
-            print("Data is not finite.")
+            if verbose:
+                print("Data is not finite.")
             out = np.zeros(location.shape[0])
         return out
 
-    def fit(self, half: int, tp: int, dt: int) -> tuple:
+    def fit(self, half: int, tp: int, dt: int, verbose: bool = False) -> tuple:
         x = np.linspace(0, 105, dt)
         y = np.linspace(0, 68, dt)
         xx, yy = np.meshgrid(x, y)
@@ -209,10 +211,10 @@ class PitchControl:
 
         for k in self.locs_home[half].keys():
             # if len(self.locs_home[half][k]) >= tp:
-            Zh += self.influence_function(k, locations, tp, 'h', half)
+            Zh += self.influence_function(k, locations, tp, 'h', half, verbose)
         for k in self.locs_away[half].keys():
             # if len(self.locs_away[half][k]) >= tp:
-            Za += self.influence_function(k, locations, tp, 'a', half)
+            Za += self.influence_function(k, locations, tp, 'a', half, verbose)
 
         Zh = Zh.reshape((dt, dt))
         Za = Za.reshape((dt, dt))

{rustat-python-api-0.5.6 → rustat-python-api-0.5.8/rustat_python_api.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustat-python-api
-Version: 0.5.6
+Version: 0.5.8
 Summary: A Python wrapper for RuStat API
 Home-page: https://github.com/dailydaniel/rustat-python-api
 Author: Daniel Zholkovsky

{rustat-python-api-0.5.6 → rustat-python-api-0.5.8}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='rustat-python-api',
-    version='0.5.6',
+    version='0.5.8',
     description='A Python wrapper for RuStat API',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',