提交 0ffd2fba authored 作者: Frederic Bastien's avatar Frederic Bastien

fix import

上级 63389092
...@@ -10,6 +10,7 @@ __docformat__ = "restructuredtext en" ...@@ -10,6 +10,7 @@ __docformat__ = "restructuredtext en"
import utils import utils
import traceback import traceback
from theano import config
class CLinkerObject(object): class CLinkerObject(object):
"""Standard elements of an Op or Type used with the CLinker """Standard elements of an Op or Type used with the CLinker
...@@ -319,7 +320,7 @@ class PureOp(object): ...@@ -319,7 +320,7 @@ class PureOp(object):
""" """
node = self.make_node(*inputs, **kwargs) node = self.make_node(*inputs, **kwargs)
limit = theano.config.traceback.limit limit = config.traceback.limit
if limit == 0: limit = None if limit == 0: limit = None
node.tag.trace = traceback.extract_stack(limit=limit)[:-1] node.tag.trace = traceback.extract_stack(limit=limit)[:-1]
if self.default_output is not None: if self.default_output is not None:
......
...@@ -7,7 +7,7 @@ import utils ...@@ -7,7 +7,7 @@ import utils
from utils import MethodNotDefined, object2 from utils import MethodNotDefined, object2
from graph import Variable from graph import Variable
import traceback import traceback
from theano import config
######## ########
# Type # # Type #
...@@ -245,7 +245,7 @@ class PureType(object): ...@@ -245,7 +245,7 @@ class PureType(object):
""" """
r = self.make_variable(name) r = self.make_variable(name)
limit = theano.config.traceback.limit limit = config.traceback.limit
if limit == 0: limit = None if limit == 0: limit = None
r.tag.trace = traceback.extract_stack(limit=limit)[:-1] r.tag.trace = traceback.extract_stack(limit=limit)[:-1]
return r return r
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论