Skip to content

Check None shape by type not identity in numba SpecifyShape dispatch#2191

Merged
jessegrabowski merged 1 commit into
pymc-devs:mainfrom
jessegrabowski:numba-specify-shape
May 30, 2026
Merged

Check None shape by type not identity in numba SpecifyShape dispatch#2191
jessegrabowski merged 1 commit into
pymc-devs:mainfrom
jessegrabowski:numba-specify-shape

Conversation

@jessegrabowski
Copy link
Copy Markdown
Member

Closes #2188

@jessegrabowski jessegrabowski requested a review from ricardoV94 May 30, 2026 03:39
@jessegrabowski jessegrabowski added bug Something isn't working numba labels May 30, 2026
Copy link
Copy Markdown
Member

@ricardoV94 ricardoV94 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is better but why did it fail without it?

@jessegrabowski
Copy link
Copy Markdown
Member Author

NoneConst is just a variable we make. Rebuilding it results in a new object because pickle doesn't carry module-level globals along for the ride. You already recognized this as a problem and cut over most things here, so this just seems like a case that got missed.

@jessegrabowski jessegrabowski merged commit c4ebafa into pymc-devs:main May 30, 2026
66 checks passed
@jessegrabowski jessegrabowski deleted the numba-specify-shape branch May 30, 2026 15:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working numba

Projects

None yet

Development

Successfully merging this pull request may close these issues.

BUG: numba_funcified_fgraph source can't be read with parallel=True

2 participants