fix: to_py_scalar was raising for decimals

narwhals-dev · Jan 11, 2025 · 1d5c57f · 1d5c57f
1 parent 346724e
commit 1d5c57f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/narwhals/translate.py b/narwhals/translate.py
@@ -2,6 +2,7 @@
 
 from datetime import datetime
 from datetime import timedelta
+from decimal import Decimal
 from functools import wraps
 from typing import TYPE_CHECKING
 from typing import Any
@@ -59,6 +60,7 @@
     int,
     float,
     complex,
+    Decimal,
 )
 
 

diff --git a/tests/translate/to_py_scalar_test.py b/tests/translate/to_py_scalar_test.py
@@ -2,6 +2,7 @@
 
 from datetime import datetime
 from datetime import timedelta
+from decimal import Decimal
 from typing import Any
 
 import numpy as np
@@ -20,6 +21,7 @@
         (1, 1),
         (pa.scalar(1), 1),
         (np.int64(1), 1),
+        (Decimal("1.1"), Decimal("1.1")),
         (1.0, 1.0),
         (None, None),
         ("a", "a"),