bayesian-sparse-gmm 0.2.2__tar.gz → 0.2.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 (39) hide show
  1. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/PKG-INFO +1 -1
  2. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/__init__.py +1 -1
  3. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/backends/_cuda.py +1 -1
  4. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/.flake8 +0 -0
  5. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/.github/workflows/release.yml +0 -0
  6. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/.gitignore +0 -0
  7. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/LICENSE +0 -0
  8. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/README.md +0 -0
  9. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/evaluate.py +0 -0
  10. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/pyproject.toml +0 -0
  11. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/backends/__init__.py +0 -0
  12. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/backends/_base.py +0 -0
  13. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/backends/_numba.py +0 -0
  14. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/backends/_numpy.py +0 -0
  15. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/config.py +0 -0
  16. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/diagnostics.py +0 -0
  17. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/model.py +0 -0
  18. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/postprocessing.py +0 -0
  19. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/sampler.py +0 -0
  20. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/state.py +0 -0
  21. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/urn.py +0 -0
  22. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/utils.py +0 -0
  23. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/__init__.py +0 -0
  24. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_backends.py +0 -0
  25. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_cuda.py +0 -0
  26. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_diagnostics.py +0 -0
  27. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_identity_cov.py +0 -0
  28. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_laplace.py +0 -0
  29. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_model.py +0 -0
  30. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_numba.py +0 -0
  31. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_phase1.py +0 -0
  32. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_sampler.py +0 -0
  33. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_synthetic.py +0 -0
  34. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_theta_update.py +0 -0
  35. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_urn_weight.py +0 -0
  36. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_variable_k.py +0 -0
  37. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_xi.py +0 -0
  38. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/tests/test_xi_update.py +0 -0
  39. {bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/uv.lock +0 -0
{bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: bayesian-sparse-gmm
3
- Version: 0.2.2
3
+ Version: 0.2.3
4
4
  Summary: Bayesian Sparse Gaussian Mixture Model implementation in Python
5
5
  Author-email: Nam Nam <nampvh4436@gmail.com>
6
6
  License: MIT
{bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/__init__.py RENAMED
@@ -1,5 +1,5 @@
1
1
  from .model import BayesianSparseGMM
2
2
 
3
- __version__ = "0.2.2"
3
+ __version__ = "0.2.3"
4
4
 
5
5
  __all__ = ["BayesianSparseGMM"]
{bayesian_sparse_gmm-0.2.2 → bayesian_sparse_gmm-0.2.3}/src/bayesian_sparse_gmm/backends/_cuda.py RENAMED
@@ -49,7 +49,7 @@ class CUDABackend(ComputeBackend):
49
49
 
50
50
  n_k = cp.bincount(z_gpu, minlength=K_max)
51
51
  sum_x = cp.zeros((K_max, X.shape[1]), dtype=X_gpu.dtype)
52
- cp.scatter_add(sum_x, z_gpu, X_gpu)
52
+ cp.add.at(sum_x, z_gpu, X_gpu)
53
53
 
54
54
  return cp.asnumpy(n_k), cp.asnumpy(sum_x)
55
55