def test_onnxt_pickle_check(self):
idi = numpy.identity(2)
onx = OnnxAdd('X', idi, output_names=['Y'])
model_def = onx.to_onnx({'X': idi.astype(numpy.float32)})
oinf = OnnxInference(model_def)
shape = oinf.shape_inference()
self.assertNotEmpty(shape)
try:
oinf.check_model()
except ValidationError as e:
warnings.warn("Why? " + str(e)) # pylint: disable=E1101
pkl = pickle.dumps(oinf)
obj = pickle.loads(pkl)
self.assertEqual(str(oinf), str(obj))
def test_onnxt_pickle_check(self):
idi = numpy.identity(2).astype(numpy.float32)
onx = OnnxAdd('X', idi, output_names=['Y'], op_version=TARGET_OPSET)
model_def = onx.to_onnx({'X': idi.astype(numpy.float32)},
target_opset=TARGET_OPSET)
oinf = OnnxInference(model_def)
shape = oinf.shape_inference()
self.assertNotEmpty(shape)
if not sys.platform.startswith('win'):
# Crashes (onnx crashes).
try:
oinf.check_model()
except ValidationError as e:
warnings.warn("Why? " + str(e)) # pylint: disable=E1101
pkl = pickle.dumps(oinf)
obj = pickle.loads(pkl)
self.assertEqual(str(oinf), str(obj))
