Skip to content

Numba doesn't respect scalar types #1407

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

Open
ricardoV94 opened this issue May 20, 2025 · 1 comment
Open

Numba doesn't respect scalar types #1407

ricardoV94 opened this issue May 20, 2025 · 1 comment
Labels
bug Something isn't working numba

Comments

@ricardoV94
Copy link
Member

Description

Example revealed in #1406

import numpy as np
import pytensor.scalar as ps

x = ps.float64("x")
y = x.astype("float32")
y.eval({x: np.int64(0.0)}, mode="NUMBA").dtype  # AttributeError: 'float' object has no attribute 'dtype'
@ricardoV94 ricardoV94 added bug Something isn't working numba labels May 20, 2025
@ricardoV94
Copy link
Member Author

As per the devs it actually respects internally, just not when returning to python. We can tweak the tests to use tensor_from_scalar perhaps?

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

No branches or pull requests

1 participant