提交 fa143f13 authored 作者: Chiheb Trabelsi's avatar Chiheb Trabelsi

type.py has been modified in order to respect the flake8 style.

上级 4fd12c42
......@@ -259,8 +259,8 @@ class CudaNdarrayType(Type):
'complex64': (complex, 'theano_complex64',
'NPY_COMPLEX64')}[self.dtype]
except KeyError:
raise TypeError("Unsupported dtype for %s: %s" % (
self.__class__.__name__, self.dtype))
raise TypeError("Unsupported dtype for %s: %s" %
(self.__class__.__name__, self.dtype))
def __eq__(self, other):
"""
......@@ -271,9 +271,10 @@ class CudaNdarrayType(Type):
other.broadcastable == self.broadcastable)
def convert_variable(self, var):
if (type(self) == type(var.type) and
if (isinstance(self, type(var.type)) and
self.ndim == var.type.ndim and
all(sb == ob or ob for sb, ob in zip(self.broadcastable,
all(sb == ob or ob for sb, ob in zip(
self.broadcastable,
var.type.broadcastable))):
return theano.tensor.patternbroadcast(var, self.broadcastable)
......@@ -312,7 +313,7 @@ class CudaNdarrayType(Type):
return self.name
else:
b = self.broadcastable
#bcast = str(self.broadcastable)
# bcast = str(self.broadcastable)
if not numpy.any(b):
s = "%iD" % len(b)
else:
......@@ -327,7 +328,7 @@ class CudaNdarrayType(Type):
def __repr__(self):
return str(self)
#"CudaNdarrayType{%s, %s}" % (str(self.dtype), str(self.broadcastable))
# "CudaNdarrayType{%s, %s}" % (str(self.dtype), str(self.broadcastable))
def c_declare(self, name, sub, check_input=True):
return """ CudaNdarray * %(name)s;""" % locals()
......@@ -563,8 +564,7 @@ theano.compile.register_deep_copy_op_c_code(
CudaNdarray_HOST_DIMS(%(oname)s)[i]) {
alloc = true;
break;
}
}
}}
if(alloc) {
Py_XDECREF(%(oname)s);
%(oname)s = (CudaNdarray*)CudaNdarray_Copy(%(iname)s);
......@@ -581,8 +581,7 @@ theano.compile.register_deep_copy_op_c_code(
%(fail)s;
}
}
""",
version=3)
""", version=3)
# THIS WORKS But CudaNdarray instances don't compare equal to one
......@@ -608,5 +607,5 @@ def CudaNdarray_pickler(cnda):
# In case cuda is not imported.
if cuda is not None:
copyreg.pickle(cuda.CudaNdarray, CudaNdarray_pickler,
CudaNdarray_unpickler)
copyreg.pickle(
cuda.CudaNdarray, CudaNdarray_pickler, CudaNdarray_unpickler)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论