PyType (@NotNull TypeEvalContext context, @NotNull TypeEvalContext.Key key) { final PyExpression truePart = getTruePart(); final PyExpression falsePart = getFalsePart(); if (truePart == null || falsePart == null) { return null; } return PyUnionType.union(context.getType(truePart), context.getType(falsePart)); }