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))