cd-dynamax 0.2.5__tar.gz → 0.3.2__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 (112) hide show
  1. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/PKG-INFO +2 -1
  2. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/__init__.py +5 -0
  3. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax.egg-info/PKG-INFO +2 -1
  4. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax.egg-info/requires.txt +1 -0
  5. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/pyproject.toml +2 -1
  6. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/LICENSE +0 -0
  7. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/README.md +0 -0
  8. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/__init__.py +0 -0
  9. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/_version.py +0 -0
  10. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/generalized_gaussian_ssm/__init__.py +0 -0
  11. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/generalized_gaussian_ssm/inference.py +0 -0
  12. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/generalized_gaussian_ssm/inference_test.py +0 -0
  13. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/generalized_gaussian_ssm/models.py +0 -0
  14. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/generalized_gaussian_ssm/models_test.py +0 -0
  15. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/__init__.py +0 -0
  16. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/inference.py +0 -0
  17. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/inference_test.py +0 -0
  18. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/__init__.py +0 -0
  19. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/abstractions.py +0 -0
  20. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/arhmm.py +0 -0
  21. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/bernoulli_hmm.py +0 -0
  22. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/categorical_glm_hmm.py +0 -0
  23. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/categorical_hmm.py +0 -0
  24. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/gamma_hmm.py +0 -0
  25. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/gaussian_hmm.py +0 -0
  26. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/gmm_hmm.py +0 -0
  27. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/initial.py +0 -0
  28. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/linreg_hmm.py +0 -0
  29. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/logreg_hmm.py +0 -0
  30. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/multinomial_hmm.py +0 -0
  31. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/poisson_hmm.py +0 -0
  32. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/test_models.py +0 -0
  33. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/models/transitions.py +0 -0
  34. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/hidden_markov_model/parallel_inference.py +0 -0
  35. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/linear_gaussian_ssm/__init__.py +0 -0
  36. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/linear_gaussian_ssm/builders.py +0 -0
  37. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/linear_gaussian_ssm/inference.py +0 -0
  38. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/linear_gaussian_ssm/inference_test.py +0 -0
  39. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/linear_gaussian_ssm/info_inference.py +0 -0
  40. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/linear_gaussian_ssm/info_inference_test.py +0 -0
  41. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/linear_gaussian_ssm/models.py +0 -0
  42. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/linear_gaussian_ssm/models_test.py +0 -0
  43. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/linear_gaussian_ssm/parallel_inference.py +0 -0
  44. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/linear_gaussian_ssm/parallel_inference_test.py +0 -0
  45. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/nonlinear_gaussian_ssm/__init__.py +0 -0
  46. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/nonlinear_gaussian_ssm/inference_ekf.py +0 -0
  47. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/nonlinear_gaussian_ssm/inference_ekf_test.py +0 -0
  48. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/nonlinear_gaussian_ssm/inference_test_utils.py +0 -0
  49. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/nonlinear_gaussian_ssm/inference_ukf.py +0 -0
  50. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/nonlinear_gaussian_ssm/inference_ukf_test.py +0 -0
  51. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/nonlinear_gaussian_ssm/models.py +0 -0
  52. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/nonlinear_gaussian_ssm/sarkka_lib.py +0 -0
  53. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/parameters.py +0 -0
  54. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/parameters_test.py +0 -0
  55. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/slds/__init__.py +0 -0
  56. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/slds/inference.py +0 -0
  57. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/slds/inference_test.py +0 -0
  58. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/slds/mixture_kalman_filter_demo.py +0 -0
  59. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/slds/models.py +0 -0
  60. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/ssm.py +0 -0
  61. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/types.py +0 -0
  62. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/utils/__init__.py +0 -0
  63. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/utils/bijectors.py +0 -0
  64. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/utils/distributions.py +0 -0
  65. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/utils/distributions_test.py +0 -0
  66. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/utils/optimize.py +0 -0
  67. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/utils/plotting.py +0 -0
  68. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/utils/utils.py +0 -0
  69. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/utils/utils_test.py +0 -0
  70. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/dynamax/warnings.py +0 -0
  71. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/__init__.py +0 -0
  72. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_linear_gaussian_ssm/__init__.py +0 -0
  73. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_linear_gaussian_ssm/cdlgssm_utils.py +0 -0
  74. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_linear_gaussian_ssm/inference.py +0 -0
  75. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_linear_gaussian_ssm/models.py +0 -0
  76. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_nonlinear_gaussian_ssm/__init__.py +0 -0
  77. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_nonlinear_gaussian_ssm/cdnlgssm_utils.py +0 -0
  78. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_nonlinear_gaussian_ssm/inference_ekf.py +0 -0
  79. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_nonlinear_gaussian_ssm/inference_enkf.py +0 -0
  80. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_nonlinear_gaussian_ssm/inference_ukf.py +0 -0
  81. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_nonlinear_gaussian_ssm/models.py +0 -0
  82. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_nonlinear_ssm/__init__.py +0 -0
  83. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_nonlinear_ssm/cdnlssm_utils.py +0 -0
  84. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_nonlinear_ssm/inference_dpf.py +0 -0
  85. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/continuous_discrete_nonlinear_ssm/models.py +0 -0
  86. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/ssm_temissions.py +0 -0
  87. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/__init__.py +0 -0
  88. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/data_driven_models.py +0 -0
  89. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/data_generator.py +0 -0
  90. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/debug_utils.py +0 -0
  91. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/demo_utils.py +0 -0
  92. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/diffrax_utils.py +0 -0
  93. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/evaluation_utils.py +0 -0
  94. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/experiment_utils.py +0 -0
  95. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/likelihood_eval_utils.py +0 -0
  96. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/optimize_utils.py +0 -0
  97. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/physics_based_models.py +0 -0
  98. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/plotting_chaos_utils.py +0 -0
  99. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/plotting_utils.py +0 -0
  100. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/prior_utils.py +0 -0
  101. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/simulation_utils.py +0 -0
  102. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/src/utils/test_utils.py +0 -0
  103. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax.egg-info/SOURCES.txt +0 -0
  104. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax.egg-info/dependency_links.txt +0 -0
  105. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax.egg-info/top_level.txt +0 -0
  106. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/setup.cfg +0 -0
  107. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/tests/test_cdlgssm_dlgssm_match.py +0 -0
  108. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/tests/test_cdnonlinear_cdlinear_match.py +0 -0
  109. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/tests/test_filter_forecast_emissions.py +0 -0
  110. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/tests/test_imports.py +0 -0
  111. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/tests/test_models.py +0 -0
  112. {cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/tests/test_utils_imports.py +0 -0
{cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cd_dynamax
3
- Version: 0.2.5
3
+ Version: 0.3.2
4
4
  Summary: Continuous-discrete dynamical systems with JAX and related libraries.
5
5
  Author: Matthew Levine, Iñigo Urteaga
6
6
  Maintainer-email: Matthew Levine <matt@basis.ai>, Iñigo Urteaga <iurteaga@bcamath.org>
@@ -29,6 +29,7 @@ Requires-Dist: distrax>=0.1.3
29
29
  Requires-Dist: orbax-checkpoint>=0.9
30
30
  Requires-Dist: etils>=1.5
31
31
  Requires-Dist: dm-tree>=0.1.8
32
+ Requires-Dist: fastprogress>=1.0.0
32
33
  Requires-Dist: graphviz
33
34
  Requires-Dist: ipykernel
34
35
  Provides-Extra: dev
{cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax/__init__.py RENAMED
@@ -1,5 +1,9 @@
1
1
  # cd_dynamax/__init__.py
2
2
 
3
+ from importlib.metadata import version
4
+
5
+ __version__ = version("cd_dynamax")
6
+
3
7
  # Nonlinear SSM
4
8
  from .src.continuous_discrete_nonlinear_gaussian_ssm import (
5
9
  ContDiscreteNonlinearGaussianSSM,
@@ -45,6 +49,7 @@ from .src.utils.optimize_utils import make_optimizer
45
49
  from .src.utils.simulation_utils import make_key_sequence
46
50
 
47
51
  __all__ = [
52
+ "__version__",
48
53
  # Models
49
54
  "ContDiscreteNonlinearGaussianSSM",
50
55
  "ContDiscreteNonlinearSSM",
{cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cd_dynamax
3
- Version: 0.2.5
3
+ Version: 0.3.2
4
4
  Summary: Continuous-discrete dynamical systems with JAX and related libraries.
5
5
  Author: Matthew Levine, Iñigo Urteaga
6
6
  Maintainer-email: Matthew Levine <matt@basis.ai>, Iñigo Urteaga <iurteaga@bcamath.org>
@@ -29,6 +29,7 @@ Requires-Dist: distrax>=0.1.3
29
29
  Requires-Dist: orbax-checkpoint>=0.9
30
30
  Requires-Dist: etils>=1.5
31
31
  Requires-Dist: dm-tree>=0.1.8
32
+ Requires-Dist: fastprogress>=1.0.0
32
33
  Requires-Dist: graphviz
33
34
  Requires-Dist: ipykernel
34
35
  Provides-Extra: dev
{cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/cd_dynamax.egg-info/requires.txt RENAMED
@@ -20,6 +20,7 @@ distrax>=0.1.3
20
20
  orbax-checkpoint>=0.9
21
21
  etils>=1.5
22
22
  dm-tree>=0.1.8
23
+ fastprogress>=1.0.0
23
24
  graphviz
24
25
  ipykernel
25
26
 
{cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/pyproject.toml RENAMED
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "cd_dynamax"
7
- version = "0.2.5"
7
+ version = "0.3.2"
8
8
  requires-python = ">=3.11"
9
9
  description = "Continuous-discrete dynamical systems with JAX and related libraries."
10
10
  readme = "README.md"
@@ -40,6 +40,7 @@ dependencies = [
40
40
  "orbax-checkpoint>=0.9",
41
41
  "etils>=1.5",
42
42
  "dm-tree>=0.1.8",
43
+ "fastprogress>=1.0.0",
43
44
  "graphviz",
44
45
  "ipykernel"
45
46
  ]
{cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/LICENSE RENAMED
File without changes
{cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/README.md RENAMED
File without changes
{cd_dynamax-0.2.5 → cd_dynamax-0.3.2}/setup.cfg RENAMED
File without changes