Commit0c2a92d

committed

fix unit test

1 parentaf88e8d commit0c2a92dCopy full SHA for 0c2a92d

File tree

-5

lines changed

-5

lines changed

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -80,9 +80,9 @@ def test_transpose(self):`
`80`	`80`	`"""`
`81`	`81`	`(`
`82`	`82`	`start(opset=19)`
	`83`	`+ .vin('X', elem_type=TensorProto.FLOAT)`
`83`	`84`	`.cst(np.array([-1, 1], dtype=np.int64))`
`84`	`85`	`.rename('r')`
`85`		`- .vin('X', elem_type=TensorProto.FLOAT)`
`86`	`86`	`.bring('X', 'r')`
`87`	`87`	`.Reshape()`
`88`	`88`	`.rename('r0_0')`
`@@ -166,9 +166,9 @@ def test_export_if(self):`
`166`	`166`	`f"""`
`167`	`167`	`(`
`168`	`168`	`start(opset=19)`
	`169`	`+ .vin('X', elem_type=TensorProto.FLOAT)`
`169`	`170`	`.cst(np.array([0.0], dtype=np.float32))`
`170`	`171`	`.rename('r')`
`171`		`- .vin('X', elem_type=TensorProto.FLOAT)`
`172`	`172`	`.bring('X')`
`173`	`173`	`.ReduceSum(keepdims=1, noop_with_empty_axes=0)`
`174`	`174`	`.rename('Xs')`
`@@ -202,9 +202,9 @@ def test_aionnxml(self):`
`202`	`202`	`"""`
`203`	`203`	`(`
`204`	`204`	`start(opset=19, opsets={'ai.onnx.ml': 3})`
	`205`	`+ .vin('X', elem_type=TensorProto.FLOAT)`
`205`	`206`	`.cst(np.array([-1, 1], dtype=np.int64))`
`206`	`207`	`.rename('r')`
`207`		`- .vin('X', elem_type=TensorProto.FLOAT)`
`208`	`208`	`.bring('X', 'r')`
`209`	`209`	`.Reshape()`
`210`	`210`	`.rename('USE')`

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -138,13 +138,13 @@ def test_transpose(self):`
`138`	`138`	`initializers = []`
`139`	`139`	`sparse_initializers = []`
`140`	`140`	`functions = []`
	`141`	`+ inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))`
`141`	`142`	`initializers.append(`
`142`	`143`	`from_array(`
`143`	`144`	`np.array([-1, 1], dtype=np.int64),`
`144`	`145`	`name='r'`
`145`	`146`	`)`
`146`	`147`	`)`
`147`		`- inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))`
`148`	`148`	`nodes.append(`
`149`	`149`	`make_node_extended(`
`150`	`150`	`'Reshape',`
`@@ -278,13 +278,13 @@ def test_aionnxml(self):`
`278`	`278`	`initializers = []`
`279`	`279`	`sparse_initializers = []`
`280`	`280`	`functions = []`
	`281`	`+ inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))`
`281`	`282`	`initializers.append(`
`282`	`283`	`from_array(`
`283`	`284`	`np.array([-1, 1], dtype=np.int64),`
`284`	`285`	`name='r'`
`285`	`286`	`)`
`286`	`287`	`)`
`287`		`- inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))`
`288`	`288`	`nodes.append(`
`289`	`289`	`make_node_extended(`
`290`	`290`	`'Reshape',`

Lines changed: 2 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -75,6 +75,8 @@ def export(self, as_str, single_line: bool = False) -> Union[str, List[str]]:`
`75`	`75`	`domain=self.proto_.domain,`
`76`	`76`	`)`
`77`	`77`	`)`
	`78`	`+elifisinstance(self.proto_,GraphProto):`
	`79`	`+rows.extend(self.emitter(EventType.BEGIN_GRAPH,name=self.proto_.name))`
`78`	`80`	`else:`
`79`	`81`	`rows.extend(`
`80`	`82`	`self.emitter(EventType.BEGIN_GRAPH,name=self.proto_.graph.name)`

Comments

(0)