pymc-devs / pymc-devs/pymc-extras
`convert_flat_trace_to_idata` throws NotImplementedError with Pathfinder and pymc inference backend
Nobody has claimed this yet.
- Dominant language
- Jupyter Notebook
- Stars
- 143
- Forks
- 91
- Avg merge
- 3d 18h
- Merged PRs (30d)
- 19
Description
I get an error message
NotImplementedError: Cannot vectorize node AdvancedSetSubtensor(AllocEmpty{dtype='float'}.0, Subtensor{:, start:}.0, MakeSlice.0, [0 4]) with input MakeSlice.0 of type slice
when trying to run pymc_extras.fit_pathfinder using pymc as the inference backend.
- This doesn't happen when I manually change the inference backend to blackjax right before calling
convert_flat_trace_to_idata. - This also doesn't happen when I remove a deterministic node
test_incomethat we are using for testing as seen on the bottom row below.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reproducing the failure through pymc_extras.fit_pathfinder with the PyMC inference backend, then inspect convert_flat_trace_to_idata and the deterministic node test_income mentioned in the report. Compare this path with the BlackJAX backend and with test_income removed; done means the reported model converts without the NotImplementedError while preserving the existing behavior.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100