提交 47cc3474 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Remove aesara.compile.ops.expandable_types

上级 cbf91122
......@@ -213,11 +213,6 @@ class DeepCopyOp(COp):
deep_copy_op = DeepCopyOp()
# List of Aesara Types that one can add an extra dimension and for which
# Scan can deal with.
expandable_types = ()
def load_back(mod, name):
__import__(mod)
import sys
......
......@@ -4,7 +4,7 @@ from collections import OrderedDict
import numpy as np
import aesara.tensor as aet
from aesara.compile import SharedVariable, ops
from aesara.compile import SharedVariable
from aesara.compile.function import function
from aesara.compile.mode import Mode
from aesara.configdefaults import config
......@@ -869,7 +869,7 @@ def scan(
new_var = safe_new(input.variable)
if getattr(input.variable, "name", None) is not None:
new_var.name = input.variable.name + "_copy"
if isinstance(new_var.type, ops.expandable_types):
if isinstance(new_var.type, TensorType):
sit_sot_inner_inputs.append(new_var)
sit_sot_scan_inputs.append(
utils.expand_empty(
......
......@@ -598,9 +598,6 @@ class TensorType(CType):
return np.dtype(self.dtype).itemsize
aesara.compile.ops.expandable_types += (TensorType,)
def values_eq_approx(
a, b, allow_remove_inf=False, allow_remove_nan=False, rtol=None, atol=None
):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论