提交 b45d4cf1 authored 作者: Lijun Xue's avatar Lijun Xue

Flake8 fix

上级 80365291
...@@ -26,16 +26,18 @@ class Test_reallocation(unittest.TestCase): ...@@ -26,16 +26,18 @@ class Test_reallocation(unittest.TestCase):
mode=m) mode=m)
output = f(1, 2) output = f(1, 2)
assert output
storage_map = f.fn.storage_map storage_map = f.fn.storage_map
def check_storage(storage_map): def check_storage(storage_map):
from theano.tensor.var import TensorConstant from theano.tensor.var import TensorConstant
for i in storage_map.keys(): for i in storage_map.keys():
if not isinstance(i, TensorConstant): if not isinstance(i, TensorConstant):
keys_copy = storage_map.keys()[:] keys_copy = storage_map.keys()[:]
keys_copy.remove(i) keys_copy.remove(i)
for o in keys_copy: for o in keys_copy:
if storage_map[i][0] and storage_map[i][0] == storage_map[o][0]: if (storage_map[i][0] and
storage_map[i][0] == storage_map[o][0]):
return [True, storage_map[o][0]] return [True, storage_map[o][0]]
return [False, None] return [False, None]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论