Skip to content

Fix some type issues related to retrieving stats from traces #6504

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Feb 7, 2023

Conversation

michaelosthege
Copy link
Member

New features

  • Statsbijection.rmap can now take incomplete input dicts.

Maintenance

  • pm.sample no longer relies on MultiTrace.get_sampler_stats that does some black magic with the fetched array.
  • Addition of type hints to stats related signatures.
  • Updated some stats related docstrings.

@michaelosthege michaelosthege added the trace-backend Traces and ArviZ stuff label Feb 5, 2023
@michaelosthege michaelosthege self-assigned this Feb 5, 2023
@michaelosthege michaelosthege changed the title Trace refactoring Fix some type issues related to retrieving stats from traces Feb 5, 2023
@codecov
Copy link

codecov bot commented Feb 5, 2023

Codecov Report

Merging #6504 (7f949ed) into main (ac9652e) will increase coverage by 0.00%.
The diff coverage is 100.00%.

Additional details and impacted files

Impacted file tree graph

@@           Coverage Diff           @@
##             main    #6504   +/-   ##
=======================================
  Coverage   94.82%   94.83%           
=======================================
  Files         148      148           
  Lines       27818    27820    +2     
=======================================
+ Hits        26379    26382    +3     
+ Misses       1439     1438    -1     
Impacted Files Coverage Δ
pymc/backends/base.py 85.40% <100.00%> (+0.42%) ⬆️
pymc/backends/ndarray.py 79.27% <100.00%> (ø)
pymc/sampling/mcmc.py 93.03% <100.00%> (-0.04%) ⬇️
pymc/step_methods/compound.py 97.45% <100.00%> (ø)
pymc/tests/step_methods/test_compound.py 100.00% <100.00%> (ø)

@michaelosthege michaelosthege marked this pull request as ready for review February 5, 2023 17:26
The mcmc module relies on the `"tune"` stat to figure out the number of
tune/draw iterations post sampling.
These changes remove reliance on any weird squeeze-cat magic.

Co-authored-by: Virgile Andreani <[email protected]>
@michaelosthege michaelosthege merged commit 9eaf1f2 into main Feb 7, 2023
@michaelosthege michaelosthege deleted the trace-refactoring branch February 7, 2023 08:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
trace-backend Traces and ArviZ stuff
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants