amplify push gives error "Unknown type" when using union or interface in schema - amplify-cli

Describe the bug I defined a schema with a query that returns a type that contains an array of a union object. (see sample below). Everything works when testing locally, no errors. When I run amplify push, everything seems just fine, I get a few minutes of messages about updating things, until the very end, and then:

Unknown type EnrollmentStats for field items. Did you forget to add the @model directive
An error occurred during the push operation: Unknown type EnrollmentStats for field items. Did you forget to add the @model directive

The funny thing is, it does update the stack... I've tried using an interface instead of a union, and the same thing happens.

Amplify CLI Version 4.38.2

To Reproduce some of my schema

type Query {
  listEnrollmentStats(
    filter: ModelEnrollmentStatsFilterInput!
  ): ModelEnrollmentStatsConnection @function(name: "myfunction")
}

type ModelEnrollmentStatsConnection {
  items: [EnrollmentStats]
}

union EnrollmentStats =
    CaseEnrollmentStats
  | LocationEnrollmentStats

Expected behavior If it works, it shouldn't give me an error message, it's very disconcerting. I'm worried that something failed to update that I don't know about yet.

Asked Oct 04 '21 00:10

estyh