Skip to content

GitLab

Unverified Commit cd387899 authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Fix a corrupted test in serializer (#4845)

parent 1246a582
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 6 deletions
test/ut/sdk/imported/_test_serializer_main.py
View file @ cd387899
...@@ -11,12 +11,9 @@ from nni.retiarii import model_wrapper ...@@ -11,12 +11,9 @@ from nni.retiarii import model_wrapper
class Net(nn.Module): class Net(nn.Module):
something = 1 something = 1
import cloudpickle
if sys.argv[1] == '0': if sys.argv[1] == '0':
cloudpickle.dump(Net, open('serialize_result.txt', 'wb')) # This could be extraordinary large on MacOS
# nni.dump(Net, fp=open('serialize_result.txt', 'w')) nni.dump(Net, fp=open('serialize_result.txt', 'w'), pickle_size_limit=16384)
else: else:
obj = cloudpickle.load(open('serialize_result.txt', 'rb')) obj = nni.load(fp=open('serialize_result.txt'))
# obj = nni.load(fp=open('serialize_result.txt'))
assert obj().something == 1 assert obj().something == 1
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment