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