提交 73ac249c authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix numpy comparison in function loading.

上级 95b6e536
......@@ -659,7 +659,7 @@ def _constructor_Function(maker, input_storage, inputs_data):
f = maker.create(input_storage, trustme = True)
assert len(f.input_storage) == len(inputs_data)
for container, x in zip(f.input_storage, inputs_data):
assert (container.data is x) or (container.data == x)
assert (container.data is x) or (container.data == x).all()
return f
copy_reg.pickle(Function, _pickle_Function)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论