rustat-python-api 0.6.3__tar.gz → 0.6.4__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 (20) hide show
  1. {rustat-python-api-0.6.3/rustat_python_api.egg-info → rustat-python-api-0.6.4}/PKG-INFO +1 -1
  2. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api/pitch_control.py +4 -2
  3. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4/rustat_python_api.egg-info}/PKG-INFO +1 -1
  4. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/setup.py +1 -1
  5. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/LICENSE +0 -0
  6. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/README.md +0 -0
  7. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/pyproject.toml +0 -0
  8. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api/__init__.py +0 -0
  9. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api/config.py +0 -0
  10. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api/kernels/__init__.py +0 -0
  11. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api/kernels/maha.py +0 -0
  12. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api/models_api.py +0 -0
  13. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api/parser.py +0 -0
  14. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api/processing.py +0 -0
  15. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api/urls.py +0 -0
  16. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api.egg-info/SOURCES.txt +0 -0
  17. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api.egg-info/dependency_links.txt +0 -0
  18. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api.egg-info/requires.txt +0 -0
  19. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/rustat_python_api.egg-info/top_level.txt +0 -0
  20. {rustat-python-api-0.6.3 → rustat-python-api-0.6.4}/setup.cfg +0 -0
{rustat-python-api-0.6.3/rustat_python_api.egg-info → rustat-python-api-0.6.4}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: rustat-python-api
3
- Version: 0.6.3
3
+ Version: 0.6.4
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.6.3 → rustat-python-api-0.6.4}/rustat_python_api/pitch_control.py RENAMED
@@ -322,7 +322,9 @@ class PitchControl:
322
322
  diff = locs.view(1, -1, 2) # (1,N,2)
323
323
  diff = diff - mu.unsqueeze(1) # (P,N,2)
324
324
 
325
- if "cuda" in device:
325
+ device = torch.device(device)
326
+
327
+ if device.type == 'cuda':
326
328
  out = triton_influence(
327
329
  mu.unsqueeze(0), Sigma_inv.unsqueeze(0),
328
330
  locs, BLOCK_N=64
@@ -391,7 +393,7 @@ class PitchControl:
391
393
  diff = locs.view(1, 1, -1, 2) # (1,1,N,2)
392
394
  diff = diff - mu.unsqueeze(2) # (F,P,N,2)
393
395
 
394
- if "cuda" in device:
396
+ if device.type == 'cuda':
395
397
  out = triton_influence(mu, Sigma_inv, locs, BLOCK_N=64) # (F,N)
396
398
 
397
399
  return out
{rustat-python-api-0.6.3 → rustat-python-api-0.6.4/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.6.3
3
+ Version: 0.6.4
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.6.3 → rustat-python-api-0.6.4}/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.6.3',
5
+ version='0.6.4',
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',