rustat-python-api 0.7.2__tar.gz → 0.7.3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (24) hide show
  1. {rustat-python-api-0.7.2/rustat_python_api.egg-info → rustat-python-api-0.7.3}/PKG-INFO +1 -1
  2. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/matching/dataloader.py +8 -1
  3. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3/rustat_python_api.egg-info}/PKG-INFO +1 -1
  4. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/setup.py +1 -1
  5. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/LICENSE +0 -0
  6. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/README.md +0 -0
  7. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/pyproject.toml +0 -0
  8. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/__init__.py +0 -0
  9. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/config.py +0 -0
  10. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/kernels/__init__.py +0 -0
  11. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/kernels/maha.py +0 -0
  12. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/matching/__init__.py +0 -0
  13. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/matching/pc_adder.py +0 -0
  14. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/matching/tr_adder.py +0 -0
  15. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/models_api.py +0 -0
  16. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/parser.py +0 -0
  17. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/pitch_control.py +0 -0
  18. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/processing.py +0 -0
  19. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/urls.py +0 -0
  20. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api.egg-info/SOURCES.txt +0 -0
  21. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api.egg-info/dependency_links.txt +0 -0
  22. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api.egg-info/requires.txt +0 -0
  23. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api.egg-info/top_level.txt +0 -0
  24. {rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/setup.cfg +0 -0
{rustat-python-api-0.7.2/rustat_python_api.egg-info → rustat-python-api-0.7.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: rustat-python-api
3
- Version: 0.7.2
3
+ Version: 0.7.3
4
4
  Summary: A Python wrapper for RuStat API
5
5
  Home-page: https://github.com/dailydaniel/rustat-python-api
6
6
  Author: Daniel Zholkovsky
{rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/rustat_python_api/matching/dataloader.py RENAMED
@@ -12,6 +12,7 @@ class MatchInferLoader:
12
12
  events: pd.DataFrame, tracking: pd.DataFrame, ball: pd.DataFrame,
13
13
  modes: list[str], rads: list[int],
14
14
  radii: list[int], cone_degrees: list[int], k_list: list[int],
15
+ device: str = "cpu", backend: str = "pt"
15
16
  ):
16
17
  self.events = events
17
18
  self.tracking = tracking
@@ -22,6 +23,9 @@ class MatchInferLoader:
22
23
  self.cone_degrees = cone_degrees
23
24
  self.k_list = k_list
24
25
 
26
+ self.device = device
27
+ self.backend = backend
28
+
25
29
  def _save_index(self):
26
30
  self.events['orig_index'] = self.events.index
27
31
 
@@ -29,7 +33,10 @@ class MatchInferLoader:
29
33
  self.events = process_events_after_loading(self.events)
30
34
 
31
35
  def _add_pc_features(self):
32
- pc_adder = PitchControlAdder(self.events, self.tracking, self.ball, device="mps", backend="pt")
36
+ pc_adder = PitchControlAdder(
37
+ self.events, self.tracking, self.ball,
38
+ device=self.device, backend=self.backend
39
+ )
33
40
  pc_adder.run(modes=self.modes, rads=self.rads)
34
41
  self.events = pc_adder.events
35
42
 
{rustat-python-api-0.7.2 → rustat-python-api-0.7.3/rustat_python_api.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: rustat-python-api
3
- Version: 0.7.2
3
+ Version: 0.7.3
4
4
  Summary: A Python wrapper for RuStat API
5
5
  Home-page: https://github.com/dailydaniel/rustat-python-api
6
6
  Author: Daniel Zholkovsky
{rustat-python-api-0.7.2 → rustat-python-api-0.7.3}/setup.py RENAMED
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
2
2
 
3
3
  setup(
4
4
  name='rustat-python-api',
5
- version='0.7.2',
5
+ version='0.7.3',
6
6
  description='A Python wrapper for RuStat API',
7
7
  long_description=open('README.md').read(),
8
8
  long_description_content_type='text/markdown',