python / python/mypy

overloads not properly evaluated with mypy 2.3.0 and numpy 2.5.1

Abierto
#21,733 3 comentarios 1 reacción 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

bug topic-overloads
Lenguaje dominante
Python
Estrellas
20.6k
Forks
3.3k
Métricas de merge de PR
Métricas de PR pendientes

Descripción

Bug Report

The final reveal_type() below should show DatetimeArray, but it is showing Any

With pyright, ty and pyrefly, it produces the correct result. May be related to #21608

To Reproduce

import numpy as np
from typing import Any, Sequence, overload, reveal_type


class DatetimeArray:
    pass


class NumpyExtensionArray:
    pass


@overload
def array(  # pyright: ignore[reportOverlappingOverload]
    data: Sequence[np.datetime64],
) -> DatetimeArray: ...
@overload
def array(data: Sequence[Any]) -> NumpyExtensionArray: ...
def array(data: Sequence[Any]) -> DatetimeArray | NumpyExtensionArray:
    return (
        DatetimeArray() if isinstance(data[0], np.datetime64) else NumpyExtensionArray()
    )


foo1 = [np.datetime64("2026-01-05 23:27:59")]
foo0 = np.datetime64("2026-01-05 23:27:59")

reveal_type(foo0)
reveal_type(foo1)
reveal_type(array(foo1))  # Should be DatetimeArray

Expected Behavior

numpy_datetimearray.py:28: note: Revealed type is "numpy.datetime64[Any]"
numpy_datetimearray.py:29: note: Revealed type is "list[numpy.datetime64[Any]]"
numpy_datetimearray.py:30: note: Revealed type is "DatetimeArray"

Actual Behavior

numpy_datetimearray.py:28: note: Revealed type is "numpy.datetime64[Any]"
numpy_datetimearray.py:29: note: Revealed type is "list[numpy.datetime64[Any]]"
numpy_datetimearray.py:30: note: Revealed type is "Any"

Your Environment

  • Mypy version used: 2.3.0
  • Mypy command-line flags: None
  • Mypy configuration options from mypy.ini (and other config files): None
  • Python version used: 3.12
  • Numpy version used: 2.5.1

Guía de contribución

Abrir la guía de contribución

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Línea de trabajo

El issue no menciona ningún archivo del repositorio ni ninguna prueba existente. Empieza guardando el reproductor proporcionado y ejecutándolo con mypy 2.3.0; después, sigue la evaluación de las sobrecargas para las dos firmas de arrays y compara el resultado con la salida esperada de reveal_type. El trabajo estará terminado cuando la llamada revele DatetimeArray y una prueba de regresión cubra este caso.

Escrito por el modelo de indexación a partir del texto del issue.

Evaluación

Stack tecnológico
numpy, python
Área
devtools, tooling
Tipo de issue
Error
Dificultad
4/5
Tiempo estimado
3-5 días
Estado de actividad
Tranquilo
Claridad
Bastante claro
Aptitud para principiantes
52/100

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.