Upgrade Aesara and Aeppl dependencies

Author: michaelosthege

.pre-commit-config.yaml

@@ -26,8 +26,8 @@ repos:
       - types-filelock
       - types-setuptools
       - arviz
-      - aesara==2.5.1
-      - aeppl==0.0.27
+      - aesara==2.6.2
+      - aeppl==0.0.28
       always_run: true
       require_serial: true
       pass_filenames: false

conda-envs/environment-dev-py37.yml

@@ -4,8 +4,8 @@ channels:
 - conda-forge
 - defaults
 dependencies:
-- aeppl=0.0.27
-- aesara=2.5.1
+- aeppl=0.0.28
+- aesara=2.6.2
 - arviz>=0.12.0
 - blas
 - cachetools>=4.2.1

conda-envs/environment-dev-py38.yml

@@ -4,8 +4,8 @@ channels:
 - conda-forge
 - defaults
 dependencies:
-- aeppl=0.0.27
-- aesara=2.5.1
+- aeppl=0.0.28
+- aesara=2.6.2
 - arviz>=0.12.0
 - blas
 - cachetools>=4.2.1

conda-envs/environment-dev-py39.yml

@@ -4,8 +4,8 @@ channels:
 - conda-forge
 - defaults
 dependencies:
-- aeppl=0.0.27
-- aesara=2.5.1
+- aeppl=0.0.28
+- aesara=2.6.2
 - arviz>=0.12.0
 - blas
 - cachetools>=4.2.1

conda-envs/environment-test-py37.yml

@@ -4,8 +4,8 @@ channels:
 - conda-forge
 - defaults
 dependencies:
-- aeppl=0.0.27
-- aesara=2.5.1
+- aeppl=0.0.28
+- aesara=2.6.2
 - arviz>=0.12.0
 - blas
 - cachetools>=4.2.1

conda-envs/environment-test-py38.yml

@@ -4,8 +4,8 @@ channels:
 - conda-forge
 - defaults
 dependencies:
-- aeppl=0.0.27
-- aesara=2.5.1
+- aeppl=0.0.28
+- aesara=2.6.2
 - arviz>=0.12.0
 - blas
 - cachetools>=4.2.1

conda-envs/environment-test-py39.yml

@@ -4,8 +4,8 @@ channels:
 - conda-forge
 - defaults
 dependencies:
-- aeppl=0.0.27
-- aesara=2.5.1
+- aeppl=0.0.28
+- aesara=2.6.2
 - arviz>=0.12.0
 - blas
 - cachetools>=4.2.1

conda-envs/windows-environment-dev-py38.yml

@@ -4,8 +4,8 @@ channels:
 - defaults
 dependencies:
  # base dependencies (see install guide for Windows)
-- aeppl=0.0.27
-- aesara=2.5.1
+- aeppl=0.0.28
+- aesara=2.6.2
 - arviz>=0.12.0
 - blas
 - cachetools>=4.2.1

conda-envs/windows-environment-test-py38.yml

@@ -4,8 +4,8 @@ channels:
 - defaults
 dependencies:
  # base dependencies (see install guide for Windows)
-- aeppl=0.0.27
-- aesara=2.5.1
+- aeppl=0.0.28
+- aesara=2.6.2
 - arviz>=0.12.0
 - blas
 - cachetools>=4.2.1

pymc/distributions/mixture.py

@@ -274,10 +274,7 @@ def rv_op(cls, weights, *components, size=None, rngs=None):
 
         # Index components and squeeze mixture dimension
         mix_out_ = at.take_along_axis(stacked_components_, mix_indexes_padded_, axis=mix_axis)
-        # There is a Aesara bug in squeeze with negative axis
-        # https://github.com/aesara-devs/aesara/issues/830
-        # this is equivalent to np.squeeze(mix_out_, axis=mix_axis)
-        mix_out_ = at.squeeze(mix_out_, axis=mix_out_.ndim + mix_axis)
+        mix_out_ = at.squeeze(mix_out_, axis=mix_axis)
 
         # Output mix_indexes rng update so that it can be updated in place
         mix_indexes_rng_next_ = mix_indexes_.owner.outputs[0]
@@ -458,7 +455,7 @@ def marginal_mixture_moment(op, rv, rng, weights, *components):
     transforms.LogExpM1,
     transforms.LogOddsTransform,
     transforms.Ordered,
-    transforms.Simplex,
+    transforms.SimplexTransform,
     transforms.SumTo1,
 )