提交 32b5c779 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make pickle assert work with non-numpy objects.

上级 792c6542
......@@ -659,7 +659,9 @@ 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).all()
assert (container.data is x) or \
(isinstance(x, numpy.ndarray) and (container.data == x).all()) or \
(container.data == x)
return f
copy_reg.pickle(Function, _pickle_Function)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论