提交 2957ed4e authored 作者: Brandon T. Willard's avatar Brandon T. Willard

Move PdbBreakpoint Op from tests to the theano package

上级 f8ce40ae
......@@ -607,7 +607,7 @@ Breakpoint during Theano function execution
-------------------------------------------
You can set a breakpoint during the execution of a Theano function with
:class:`PdbBreakpoint <tests.breakpoint.PdbBreakpoint>`.
:class:`PdbBreakpoint <tests.breakpoint.PdbBreakpoint>` automatically
:class:`PdbBreakpoint <theano.breakpoint.PdbBreakpoint>`.
:class:`PdbBreakpoint <theano.breakpoint.PdbBreakpoint>` automatically
detects available debuggers and uses the first available in the following order:
`pudb`, `ipdb`, or `pdb`.
......@@ -9,6 +9,7 @@ import theano.tensor.slinalg as slinalg
from theano import tensor
from theano.gof.opt import check_stack_trace
from theano.tensor.nnet import abstract_conv
from theano.breakpoint import PdbBreakpoint
from theano.gpuarray import basic_ops
from theano.gpuarray.type import GpuArrayType, gpuarray_shared_constructor, get_context
from theano.gpuarray.basic_ops import (
......@@ -34,7 +35,6 @@ from theano.gpuarray.linalg import GpuCusolverSolve, cusolver_available, GpuChol
from tests import unittest_tools as utt, test_ifelse
from tests.tensor import test_basic
from tests.breakpoint import PdbBreakpoint
from tests.gpuarray.config import mode_with_gpu, mode_without_gpu, test_ctx_name
......
......@@ -3,8 +3,9 @@ import numpy as np
import theano
import theano.tensor as T
from theano.breakpoint import PdbBreakpoint
from tests import unittest_tools as utt
from tests.breakpoint import PdbBreakpoint
class TestPdbBreakpoint(utt.InferShapeTester):
......
......@@ -36,7 +36,7 @@ class PdbBreakpoint(Op):
import theano
import theano.tensor as T
from tests.breakpoint import PdbBreakpoint
from theano.breakpoint import PdbBreakpoint
input = T.fvector()
target = T.fvector()
......
......@@ -51,7 +51,7 @@ import theano.tensor.signal.pool as pool
import theano.tensor.slinalg as slinalg
from collections import Counter
from tests.breakpoint import PdbBreakpoint
from theano.breakpoint import PdbBreakpoint
from .type import (
GpuArrayType,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论