-_.{}[]
AssertionError: Submitted predictions are expected to have a shape of (1006, 334, 841), but found a shape of (1006, 336, 844)