Updated from OpenAPI spec

internal/mlflow_test.go

@@ -91,8 +91,13 @@ func TestAccModels(t *testing.T) {
 		deleteModel(t, w, ctx, created)
 	})
 
+	model, err := w.ModelRegistry.GetModel(ctx, ml.GetModelRequest{
+		Name: created.RegisteredModel.Name,
+	})
+	require.NoError(t, err)
+
 	err = w.ModelRegistry.UpdateModel(ctx, ml.UpdateModelRequest{
-		Name:        created.RegisteredModel.Name,
+		Name:        model.RegisteredModelDatabricks.Name,
 		Description: RandomName("comment "),
 	})
 	require.NoError(t, err)
@@ -147,6 +152,7 @@ func deleteModelVersion(t *testing.T, w *databricks.WorkspaceClient, ctx context
 		})
 		require.NoError(t, err)
 
+		// TODO: x-databricks-wait to be added to all relevant MLflow APIs
 		if currentModelVersion.ModelVersion.Status != ml.ModelVersionStatusPendingRegistration {
 			break
 		}

openapi/code/package.go

@@ -368,10 +368,10 @@ func (pkg *Package) Load(spec *openapi.Specification, tag *openapi.Tag) error {
 			seenParams := map[string]bool{}
 			for _, list := range [][]openapi.Parameter{path.Parameters, op.Parameters} {
 				for _, v := range list {
-					if v.In == "header" {
+					param, err := pkg.resolveParam(&v)
+					if param.In == "header" {
 						continue
 					}
-					param, err := pkg.resolveParam(&v)
 					if err != nil {
 						return fmt.Errorf("no components found: %s %s", verb, prefix)
 					}