Remove unused objects and functions from aesara.compile.function.types

aesara/compile/__init__.py

@@ -6,14 +6,12 @@ from aesara.compile.function.types import (
     Supervisor,
     UnusedInputError,
     alias_root,
-    check_equal,
     convert_function_input,
     fgraph_updated_vars,
     get_info_on_inputs,
     infer_reuse_pattern,
     insert_deepcopy,
     orig_function,
-    register_checker,
     std_fgraph,
     view_tree_set,
 )

aesara/compile/function/types.py

@@ -9,7 +9,6 @@ import logging
 import time
 import warnings
 from itertools import chain
-from typing import List
 
 import numpy as np
 
@@ -36,8 +35,6 @@ from aesara.link.utils import raise_with_op
 
 _logger = logging.getLogger("aesara.compile.function.types")
 
-__docformat__ = "restructuredtext en"
-
 
 class UnusedInputError(Exception):
     """
@@ -220,10 +217,6 @@ class AliasedMemoryError(Exception):
     """
 
 
-###
-# Function
-###
-
 # unique id object used as a placeholder for duplicate entries
 DUPLICATE = ["DUPLICATE"]
 
@@ -1173,9 +1166,6 @@ def _constructor_Function(maker, input_storage, inputs_data, trust_input=False):
 copyreg.pickle(Function, _pickle_Function)
 
 
-###
-# FunctionMaker
-###
 def insert_deepcopy(fgraph, wrapped_inputs, wrapped_outputs):
     """
     Insert deepcopy in the fgraph to break aliasing of outputs
@@ -1276,9 +1266,6 @@ def insert_deepcopy(fgraph, wrapped_inputs, wrapped_outputs):
                         break
 
 
-NODEFAULT = ["NODEFAULT"]
-
-
 class FunctionMaker:
     """
     `FunctionMaker` is the class to `create` `Function` instances.
@@ -1682,33 +1669,6 @@ class FunctionMaker:
         return fn
 
 
-def _constructor_FunctionMaker(kwargs):
-    # Needed for old pickle
-    # Old pickle have at least the problem that output_keys where not saved.
-    if config.unpickle_function:
-        if config.reoptimize_unpickled_function:
-            del kwargs["fgraph"]
-        return FunctionMaker(**kwargs)
-    else:
-        return None
-
-
-__checkers: List = []
-
-
-def check_equal(x, y):
-    for checker in __checkers:
-        try:
-            return checker(x, y)
-        except Exception:
-            continue
-    return x == y
-
-
-def register_checker(checker):
-    __checkers.insert(0, checker)
-
-
 def orig_function(
     inputs,
     outputs,

aesara/tensor/basic.py

@@ -68,28 +68,6 @@ _logger = logging.getLogger("aesara.tensor.basic")
 __docformat__ = "restructuredtext en"
 
 
-def check_equal_numpy(x, y):
-    """
-    Return True iff x and y are equal.
-
-    Checks the dtype and shape if x and y are numpy.ndarray instances.
-
-    """
-    if isinstance(x, np.ndarray) and isinstance(y, np.ndarray):
-        return x.dtype == y.dtype and x.shape == y.shape and np.all(abs(x - y) < 1e-10)
-    elif isinstance(x, (np.random.Generator, np.random.RandomState)) and isinstance(
-        y, (np.random.Generator, np.random.RandomState)
-    ):
-        return builtins.all(
-            np.all(a == b) for a, b in zip(x.__getstate__(), y.__getstate__())
-        )
-    else:
-        return x == y
-
-
-compile.register_checker(check_equal_numpy)
-
-
 def __oplist_tag(thing, tag):
     tags = getattr(thing, "__oplist_tags", [])
     tags.append(tag)