@Override
  public final @Nullable PyType getContextManagerVariableType(PyClass contextManager, PyExpression withExpression, TypeEvalContext context) {
    return withCustomContext(context, customContext -> {
      return getContextManagerVariableType(contextManager, withExpression, customContext);
    });
  }
@Override
  public final @Nullable PyType getContextManagerVariableType(PyClass contextManager, PyExpression withExpression, TypeEvalContext context)