rustat-python-api 0.7.4__tar.gz → 0.7.6__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.4/rustat_python_api.egg-info → rustat-python-api-0.7.6}/PKG-INFO +3 -1
  2. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/matching/dataloader.py +1 -1
  3. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6/rustat_python_api.egg-info}/PKG-INFO +3 -1
  4. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api.egg-info/requires.txt +9 -0
  5. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/setup.py +10 -1
  6. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/LICENSE +0 -0
  7. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/README.md +0 -0
  8. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/pyproject.toml +0 -0
  9. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/__init__.py +0 -0
  10. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/config.py +0 -0
  11. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/kernels/__init__.py +0 -0
  12. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/kernels/maha.py +0 -0
  13. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/matching/__init__.py +0 -0
  14. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/matching/pc_adder.py +0 -0
  15. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/matching/tr_adder.py +0 -0
  16. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/models_api.py +0 -0
  17. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/parser.py +0 -0
  18. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/pitch_control.py +0 -0
  19. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/processing.py +0 -0
  20. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/urls.py +0 -0
  21. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api.egg-info/SOURCES.txt +0 -0
  22. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api.egg-info/dependency_links.txt +0 -0
  23. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api.egg-info/top_level.txt +0 -0
  24. {rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/setup.cfg +0 -0
{rustat-python-api-0.7.4/rustat_python_api.egg-info → rustat-python-api-0.7.6}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: rustat-python-api
3
- Version: 0.7.4
3
+ Version: 0.7.6
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
@@ -11,6 +11,8 @@ Classifier: License :: OSI Approved :: MIT License
11
11
  Classifier: Operating System :: OS Independent
12
12
  Requires-Python: >=3.10
13
13
  Description-Content-Type: text/markdown
14
+ Provides-Extra: gpu
15
+ Provides-Extra: cpu
14
16
  License-File: LICENSE
15
17
 
16
18
  # rustat-python-api
{rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api/matching/dataloader.py RENAMED
@@ -50,7 +50,7 @@ class MatchInferLoader:
50
50
  self.events = tr_adder.events
51
51
 
52
52
  def fit(self, inplace: bool = False):
53
- self._save_events()
53
+ self._save_index()
54
54
  self._process_events()
55
55
  self._add_pc_features()
56
56
  self._add_tr_features()
{rustat-python-api-0.7.4 → rustat-python-api-0.7.6/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.4
3
+ Version: 0.7.6
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
@@ -11,6 +11,8 @@ Classifier: License :: OSI Approved :: MIT License
11
11
  Classifier: Operating System :: OS Independent
12
12
  Requires-Python: >=3.10
13
13
  Description-Content-Type: text/markdown
14
+ Provides-Extra: gpu
15
+ Provides-Extra: cpu
14
16
  License-File: LICENSE
15
17
 
16
18
  # rustat-python-api
{rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/rustat_python_api.egg-info/requires.txt RENAMED
@@ -8,3 +8,12 @@ torch
8
8
 
9
9
  [:platform_system == "Linux"]
10
10
  triton==3.0.0
11
+
12
+ [cpu]
13
+ torch
14
+
15
+ [gpu]
16
+ torch
17
+
18
+ [gpu:platform_system == "Linux"]
19
+ triton==3.0.0
{rustat-python-api-0.7.4 → rustat-python-api-0.7.6}/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.4',
5
+ version='0.7.6',
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',
@@ -21,6 +21,15 @@ setup(
21
21
  'torch',
22
22
  'triton==3.0.0; platform_system=="Linux"'
23
23
  ],
24
+ extras_require={
25
+ "gpu": [
26
+ "torch",
27
+ "triton==3.0.0; platform_system=='Linux'",
28
+ ],
29
+ "cpu": [
30
+ "torch"
31
+ ]
32
+ },
24
33
  classifiers=[
25
34
  'Programming Language :: Python :: 3',
26
35
  'License :: OSI Approved :: MIT License',