diff --git a/aesara/link/numba/dispatch/scalar.py b/aesara/link/numba/dispatch/scalar.py index 28031ea988..f9dcb08a36 100644 --- a/aesara/link/numba/dispatch/scalar.py +++ b/aesara/link/numba/dispatch/scalar.py @@ -221,6 +221,7 @@ def clip(_x, _min, _max): @numba_funcify.register(Composite) def numba_funcify_Composite(op, node, **kwargs): signature = create_numba_signature(node, force_scalar=True) + _ = kwargs.pop("storage_map", None) composite_fn = numba_basic.numba_njit(signature, fastmath=config.numba__fastmath)( numba_funcify(op.fgraph, squeeze_output=True, **kwargs) )