Allow extract int from str anyvalue

pola-rs · Jan 19, 2024 · 601cd5a · 601cd5a
1 parent 1d434cc
commit 601cd5a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/crates/polars-core/src/datatypes/any_value.rs b/crates/polars-core/src/datatypes/any_value.rs
@@ -415,6 +415,7 @@ impl<'a> AnyValue<'a> {
                     NumCast::from(0)
                 }
             },
+            String(v) =>  NumCast::from((*v).parse::<i64>().ok()?),
             _ => None,
         }
     }

diff --git a/py-polars/tests/unit/dataframe/test_df.py b/py-polars/tests/unit/dataframe/test_df.py
@@ -3369,6 +3369,13 @@ def test_from_dicts_undeclared_column_dtype() -> None:
     assert result.schema == {"x": pl.Null}
 
 
+def test_from_dicts_with_override() -> None:
+    data = [{'a': '1', 'b': '2'}, {'a':1, 'b':2}]
+    override = {'a': pl.Int32, 'b': pl.UInt8}
+    result = pl.from_dicts(data, schema_overrides=override)
+    assert result.schema == {"a": pl.Int32, "b": pl.UInt8}
+
+
 def test_from_records_u64_12329() -> None:
     s = pl.from_records([{"a": 9908227375760408577}])
     assert s.dtypes == [pl.UInt64]