fix: Incorrect request body for create and patch version

[tiopramayudi]

Description

Previously in this #518 we introduce CRUD API for model schema, also changed the version schema that aim to update model schema during version creation. While CRUID API is working correctly, create model version with model schema info is not working properly, due to the request body type that is used for the controller is not Version but VersionPost. Hence this PR try to fix that issue

Modifications

• api/api/version_api.go -> Update the model version creation by supplying model schema data from VersionPost to Version
• api/models/version.go -> Including ModelSchema field for VersionPost and VersionPatch struct to supply model schema info during model version creation or patch
• python/sdk/merlin/model_schema.py -> Add output_class field to model prediction output
• python/sdk/test/integration_test.py -> Add new integration test for model schema

Tests

Checklist

• Added PR label
• Added unit test, integration, and/or e2e tests
• Tested locally
• Updated documentation
• Update Swagger spec if the PR introduce API changes
• Regenerated Golang and Python client if the PR introduces API changes

Release Notes

[ghost]

👇 Click on the image for a new way to code review

Legend

[khorshuheng]

Seems like the deserialization failed because fields in RegressionOutput is a subset of the fields in BinaryClassification. Since extra fields are ignored, both passed the validation test and result in more than one match.

[khorshuheng]

I checked the swagger schema. Are we missing the mapping here?

    ModelPredictionOutput:
      type: object
      discriminator:
        propertyName: output_class
        mapping:
          BinaryClassificationOutput: '#/components/schemas/BinaryClassificationOutput'
          RankingOutput: '#/components/schemas/RankingOutput'

[tiopramayudi]

I checked the swagger schema. Are we missing the mapping here?

    ModelPredictionOutput:
      type: object
      discriminator:
        propertyName: output_class
        mapping:
          BinaryClassificationOutput: '#/components/schemas/BinaryClassificationOutput'
          RankingOutput: '#/components/schemas/RankingOutput'

this is due to not using discriminator lookup, hence they try to deserialize to all possible schema