FIX : BUG: Float Nullable dtypes not being returned when use_nullable_dtypes=True in pd.read_parquet pandas-dev#45694

Jaafarben2 · Jaafarben2 · commit 025c5fdb57a4 · 2022-01-29T22:18:05.000+01:00
Add float types in mapping dictionary
diff --git a/pandas/io/parquet.py b/pandas/io/parquet.py
@@ -231,6 +231,8 @@ def read(
                 self.api.uint64(): pd.UInt64Dtype(),
                 self.api.bool_(): pd.BooleanDtype(),
                 self.api.string(): pd.StringDtype(),
+                self.api.float32(): pd.Float32Dtype(),
+                self.api.float64(): pd.Float64Dtype(),
             }
             to_pandas_kwargs["types_mapper"] = mapping.get
         manager = get_option("mode.data_manager")

Original file line number	Diff line number	Diff line change
`@@ -231,6 +231,8 @@ def read(`
`231`	`231`	`self.api.uint64(): pd.UInt64Dtype(),`
`232`	`232`	`self.api.bool_(): pd.BooleanDtype(),`
`233`	`233`	`self.api.string(): pd.StringDtype(),`
	`234`	`+ self.api.float32(): pd.Float32Dtype(),`
	`235`	`+ self.api.float64(): pd.Float64Dtype(),`
`234`	`236`	`}`
`235`	`237`	`to_pandas_kwargs["types_mapper"] = mapping.get`
`236`	`238`	`manager = get_option("mode.data_manager")`