Switch to setup-micromamba action

[maresb]

Description

Related Issue

• Closes Bump conda-incubator/setup-miniconda from 2 to 3 #7046
• Related to #

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify):

---

📚 Documentation preview 📚: https://pymc--7213.org.readthedocs.build/en/7213/

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 92.84%. Comparing base (c17491d) to head (97dc1ca).
Report is 3 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #7213   +/-   ##
=======================================
  Coverage   92.84%   92.84%           
=======================================
  Files         106      106           
  Lines       17685    17685           
=======================================
+ Hits        16419    16420    +1     
+ Misses       1266     1265    -1     

see 1 file with indirect coverage changes

---- 🚨 Try these New Features:

• Flaky Tests Detection - Detect and resolve failed and flaky tests

[maresb]

I think the error may be fixed here: pymc-devs/pytensor#678

[maresb]

Ah ya, thanks, that's the one! (Wrong PR)

[twiecki]

Looks much simpler.

[ricardoV94]

Do we need pymc-devs/pytensor#678 here?

[twiecki]

Worth a try.

[maresb]

Ok, rebased.

[maresb]

That's weird, maybe it needs to be rerun?

I still find GH's UI here weird. I can't find a rerun button for that task. Maybe I need to wait until the whole thing finishes?

[ricardoV94]

Yeah, you can only rerun failed runs once everything is done

[maresb]

Just started the rerun. Any idea why I can't view logs (see screenshot)? Is this a GH outage?

[maresb]

Viewing the CI logs is still really flaky for me, but somehow I was able to get a glimpse of:

 _________________ ERROR collecting tests/sampling/test_jax.py __________________
tests/sampling/test_jax.py:30: in <module>
    from numpyro.infer import MCMC
../../../micromamba/envs/pymc-test/lib/python3.12/site-packages/numpyro/__init__.py:6: in <module>
    from numpyro import compat, diagnostics, distributions, handlers, infer, ops, optim
../../../micromamba/envs/pymc-test/lib/python3.12/site-packages/numpyro/distributions/__init__.py:4: in <module>
    from numpyro.distributions.conjugate import (
../../../micromamba/envs/pymc-test/lib/python3.12/site-packages/numpyro/distributions/conjugate.py:9: in <module>
    from numpyro.distributions.continuous import Beta, Dirichlet, Gamma
../../../micromamba/envs/pymc-test/lib/python3.12/site-packages/numpyro/distributions/continuous.py:36: in <module>
    from jax.scipy.linalg import cho_solve, solve_triangular
../../../micromamba/envs/pymc-test/lib/python3.12/site-packages/jax/scipy/linalg.py:18: in <module>
    from jax._src.scipy.linalg import (
../../../micromamba/envs/pymc-test/lib/python3.12/site-packages/jax/_src/scipy/linalg.py:403: in <module>
    @_wraps(scipy.linalg.tril)
E   AttributeError: module 'scipy.linalg' has no attribute 'tril'

Then for attempt #2, which I believe was post-rebase, I think I was still seeing the old "filename too long" error. 😢 But I can't access the logs in order to paste them here...

Am I the only one who is having trouble viewing the CI logs?

[maresb]

Regarding scipy.linalg.tril, see https://docs.scipy.org/doc/scipy-1.11.4/reference/generated/scipy.linalg.tril.html:

Deprecated since version 1.11.0: tril is deprecated in favour of numpy.tril and will be removed in SciPy 1.13.0.

[maresb]

Yeah, you can only rerun failed runs once everything is done

@ricardoV94, here I at least have a "Re-run" link, even though it doesn't appear to have actually triggered a rerun yet.

[ricardoV94]

@maresb wanna try to revive this one?

[maresb]

Eventually! Thanks @ricardoV94 for the ping. Currently I'm waiting for all the the major micromamba v2 bugs to get resolved before touching this sort of thing.

[maresb]

@lucianopaz, would you be interested in helping me revive this?

There are still issues with micromamba v2, but I think there's a decent chance that they won't be problematic here.

[maresb]

Supersedes #7580

[lucianopaz]

LGTM!