@Nullable
    private PyType getExpectedReturnType(@NotNull PyFunction function) {
      return getExpectedReturnType(function, myTypeEvalContext);
    }
@Nullable
    private PyType getExpectedReturnType(@NotNull PyFunction function)