Skip to content

Conversation

@kere-nel
Copy link
Contributor

Fix a small bug where the source variable name of an ONNX graph was mistakenly always the input variable name, which is not always the case. Other ONNX transformers don't explicitly test for this functionality, so I didn't add a test case.

@kere-nel kere-nel requested a review from a team as a code owner February 20, 2020 22:27
string srcVariableName = ctx.GetVariableName(inputColumnName);
var dstVariableName = ctx.AddIntermediateVariable(_types[iinfo], info.outputColumnName, true);
if (!_kvMaps[iinfo].SaveOnnx(ctx, inputColumnName, dstVariableName))
if (!_kvMaps[iinfo].SaveOnnx(ctx, srcVariableName, dstVariableName))
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think it would be a good idea to add a test case specifically because we don't have tests for this. Also please verify this fixes the Nimbus issue.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sounds good. I restructured the tests and got rid of a KeyToValue test that seems redundant and is not formatted to test different types of values.

@harishsk harishsk merged commit 275f4c2 into dotnet:master Feb 21, 2020
@ghost ghost locked as resolved and limited conversation to collaborators Mar 19, 2022
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants