File tree Expand file tree Collapse file tree 1 file changed +13
-1
lines changed
pytensor/link/numba/dispatch Expand file tree Collapse file tree 1 file changed +13
-1
lines changed Original file line number Diff line number Diff line change 55
66import numba
77import numpy as np
8- from numba import NumbaPerformanceWarning
8+ from numba import NumbaPerformanceWarning , NumbaWarning
99from numba import njit as _njit
1010from numba .cpython .unsafe .tuple import tuple_setitem # noqa: F401
1111
3333 category = NumbaPerformanceWarning ,
3434)
3535
36+ # Suppress large global arrays cache warning for internal functions
37+ # We have to add an ansi escape code for optional bold text by numba
38+ warnings .filterwarnings (
39+ "ignore" ,
40+ message = (
41+ "(\x1b \\ [1m)*" # ansi escape code for bold text
42+ 'Cannot cache compiled function "numba_funcified_fgraph" '
43+ "as it uses dynamic globals"
44+ ),
45+ category = NumbaWarning ,
46+ )
47+
3648
3749def numba_njit (
3850 * args , fastmath = None , final_function : bool = False , ** kwargs
You can’t perform that action at this time.
0 commit comments