提交 813d0409 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Revert to lazy loaded aesara.link.c.cutils

上级 6cbb51c2
......@@ -21,7 +21,6 @@ from aesara.graph.basic import (
vars_between,
)
from aesara.link.basic import Container, Linker, LocalLinker, PerformLinker
from aesara.link.c import cutils
from aesara.link.c.cmodule import (
METH_VARARGS,
DynamicModule,
......@@ -1722,7 +1721,7 @@ class _CThunk:
Parameters
----------
cthunk
The CObject pointer used by run_cthunk.
A CObject pointer that is used to run the thunk.
init_tasks
WRITEME
tasks
......@@ -1736,6 +1735,10 @@ class _CThunk:
"""
def __init__(self, cthunk, init_tasks, tasks, error_storage, module):
# Lazy import to avoid compilation when importing aesara.
from aesara.link.c.cutils import run_cthunk # noqa
self.run_cthunk = run_cthunk
self.cthunk = cthunk
self.init_tasks = init_tasks
self.tasks = tasks
......@@ -1757,7 +1760,7 @@ class _CThunk:
return self.tasks[failure_code - n]
def __call__(self):
failure = cutils.run_cthunk(self.cthunk)
failure = self.run_cthunk(self.cthunk)
if failure:
task, taskname, id = self.find_task(failure)
try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论