Polars: infer schema for scan_parquet/read_parquet with schema=

Abierto
#5,433 2 comentarios 0 reacciones 1 asignado Ver en GitHub

@Gizzio ya está trabajando en esto.

Desde el 30/10/2026.

Evaluación

Dificultad
3/5
Tiempo estimado
1-2 días
Aptitud para principiantes
62/100
Tipo de issue
Error
Claridad
Bien especificado
Estado de actividad
Activo
Stack tecnológico
python, rust

Línea de trabajo

Start by comparing how Pyrefly infers schemas for Polars read_csv/scan_csv with its handling of read_parquet/scan_parquet when schema= is provided. The issue gives no source files or tests, so first locate the Polars API typing or inference tests and run the relevant cases. Done when both Parquet calls reveal frame types containing the supplied column names and types.

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

Descripción

dataframe
Describe the Bug

Polars schema inference works for pl.read_csv / pl.scan_csv with an explicit schema=, but not for pl.read_parquet / pl.scan_parquet with the same argument. Both functions accept schema= and the resulting schema is known statically, so I'd expect the same inference.

from typing import reveal_type
import polars as pl

reveal_type(pl.scan_csv("x.csv", schema={"vehicle": pl.String, "node": pl.UInt8}))
# LazyFrame[vehicle: String, node: UInt8]   <- OK

reveal_type(pl.scan_parquet("x/", schema={"vehicle": pl.String, "node": pl.UInt8}))
# LazyFrame                                 <- expected LazyFrame[vehicle: String, node: UInt8]

reveal_type(pl.read_parquet("x/", schema={"vehicle": pl.String, "node": pl.UInt8}))
# DataFrame                                 <- expected DataFrame[vehicle: String, node: UInt8]

Expected

scan_parquet / read_parquet with schema= infer the frame schema the same way scan_csv / read_csv do.

Environment

  • pyrefly 1.4.0
  • polars 2.1.0
  • Python 3.14

Related: #5036 (read_csv with schema_overrides), #5271 (annotated lazy frames).

Sandbox Link

No response

(Only applicable for extension issues) IDE Information

No response

Lenguaje dominante
Rust
Estrellas
7.2k
Forks
578
Métricas de merge de PR
Sin PR fusionados en 30 d

Preparar el entorno

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.

Más de facebook/pyrefly

Todos los issues de facebook/pyrefly

Issues similares

Más issues de Rust

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.