Skip to content

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