提交 2c16bfe7 authored 作者: Bart van Merriënboer's avatar Bart van Merriënboer

Fix error and remove unnecessary constructor

上级 87f8288b
......@@ -153,9 +153,6 @@ class PersistentNdarrayID(object):
class PersistentCudaNdarrayID(PersistentNdarrayID):
def __init__(self, zip_file):
super(PersistentCudaNdarrayID, self).__init__(zip_file)
def __call__(self, obj):
if (cuda_ndarray is not None and
type(obj) is cuda_ndarray.cuda_ndarray.CudaNdarray):
......@@ -203,12 +200,12 @@ class PersistentSharedVariableID(PersistentCudaNdarrayID):
if id(obj) in self.ndarray_names:
name = self.ndarray_names[id(obj)]
count = self.name_counter[name]
self.name_counter[name] += 1
if count:
if not self.allow_duplicates:
raise ValueError("multiple shared variables with the name "
"`{0}` found".format(name))
name = '{0}_{1}'.format(name, count + 1)
self.name_counter[name] += 1
return name
return super(PersistentSharedVariableID, self)._resolve_name(obj)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论