Skip to content

Commit 67c4c6c

Browse files
committed
Temporarily remove numba object mode warning
1 parent 9b1abf1 commit 67c4c6c

File tree

2 files changed

+5
-7
lines changed

2 files changed

+5
-7
lines changed

pytensor/link/numba/dispatch/basic.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -255,11 +255,11 @@ def numba_typify(data, dtype=None, **kwargs):
255255
def generate_fallback_impl(op, node, storage_map=None, **kwargs):
256256
"""Create a Numba compatible function from a Pytensor `Op`."""
257257

258-
warnings.warn(
259-
f"Numba will use object mode to run {op}'s perform method. "
260-
f"Set `pytensor.config.compiler_verbose = True` to see more details.",
261-
UserWarning,
262-
)
258+
# warnings.warn(
259+
# f"Numba will use object mode to run {op}'s perform method. "
260+
# f"Set `pytensor.config.compiler_verbose = True` to see more details.",
261+
# UserWarning,
262+
# )
263263

264264
if config.compiler_verbose:
265265
node.dprint(depth=5, print_type=True)

tests/tensor/test_basic.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,8 +151,6 @@
151151
)
152152

153153

154-
pytestmark = pytest.mark.filterwarnings("error")
155-
156154
if config.mode == "FAST_COMPILE":
157155
mode_opt = "FAST_RUN"
158156
else:

0 commit comments

Comments
 (0)