.saner defaults

Author: ricardoV94

pytensor/link/numba/cache.py

@@ -11,7 +11,6 @@
 from numba.core.caching import CacheImpl, _CacheLocator
 
 from pytensor import config
-from pytensor.graph.basic import Apply
 from pytensor.link.numba.compile import numba_funcify, numba_njit
 
 
@@ -65,22 +64,6 @@ def from_function(cls, py_func, py_file):
 CacheImpl._locator_classes.insert(0, NumbaPyTensorCacheLocator)
 
 
-def cache_node_key(node: Apply, extra_key="") -> str:
-    op = node.op
-    return sha256(
-        str(
-            (
-                # Op signature
-                (type(op), op._props_dict() if hasattr(op, "_props_dict") else ""),
-                # Node signature
-                tuple((type(inp_type := inp.type), inp_type) for inp in node.inputs),
-                # Extra key given by the caller
-                extra_key,
-            ),
-        ).encode()
-    ).hexdigest()
-
-
 @singledispatch
 def numba_funcify_default_op_cache_key(
     op, node=None, **kwargs
@@ -109,7 +92,7 @@ def decorator(dispatch_func):
         # Create a wrapper for the non-cache dispatcher
         @wraps(dispatch_func)
         def dispatch_func_wrapper(*args, **kwargs):
-            func, key = dispatch_func(*args, **kwargs)
+            func, _key = dispatch_func(*args, **kwargs)
             # Discard the key for the non-cache version
             return func
 
@@ -172,7 +155,7 @@ def decorator(dispatch_func):
         # Create a wrapper for the non-cache dispatcher
         @wraps(dispatch_func)
         def dispatch_func_wrapper(*args, **kwargs):
-            func, key = dispatch_func(*args, **kwargs)
+            func, _key = dispatch_func(*args, **kwargs)
             # Discard the key for the non-cache version
             return func