rustat-python-api 0.6.0__tar.gz → 0.6.1__tar.gz

{rustat-python-api-0.6.0/rustat_python_api.egg-info → rustat-python-api-0.6.1}/PKG-INFO

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

{rustat-python-api-0.6.0 → rustat-python-api-0.6.1}/rustat_python_api/pitch_control.py

@@ -398,11 +398,24 @@ class PitchControl:
     @staticmethod
     def _stack_team_frames(players: list[np.ndarray], frames: np.ndarray, device: str, dtype: torch.dtype):
         """Stack positions for given frames into torch tensors (pos_t, pos_tp1)."""
+        # Ensure every player's trajectory is long enough; if not, pad by repeating
+        # the last available coordinate so that indexing `frames` and `frames+1` is safe.
+        max_needed = frames[-1] + 1  # we access idx and idx+1
+
+        padded = []
+        for p in players:
+            if len(p) <= max_needed:
+                pad_len = max_needed + 1 - len(p)
+                if pad_len > 0:
+                    last = p[-1][None, :]
+                    p = np.vstack([p, np.repeat(last, pad_len, axis=0)])
+            padded.append(p)
+
         pos_t = torch.tensor(
-            np.stack([p[frames] for p in players], axis=1), device=device, dtype=dtype
+            np.stack([p[frames] for p in padded], axis=1), device=device, dtype=dtype
         )  # (F,P,2)
         pos_tp1 = torch.tensor(
-            np.stack([p[frames + 1] for p in players], axis=1), device=device, dtype=dtype
+            np.stack([p[frames + 1] for p in padded], axis=1), device=device, dtype=dtype
         )
         return pos_t, pos_tp1
 

{rustat-python-api-0.6.0 → rustat-python-api-0.6.1/rustat_python_api.egg-info}/PKG-INFO

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

{rustat-python-api-0.6.0 → rustat-python-api-0.6.1}/setup.py

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