Allow fill_null to be used with torch tensors

pola-rs · Oct 7, 2024 · 28e7fc7 · 28e7fc7
1 parent addaf83
commit 28e7fc7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py-polars/polars/functions/lit.py b/py-polars/polars/functions/lit.py
@@ -158,7 +158,7 @@ def lit(
     if dtype:
         return wrap_expr(plr.lit(value, allow_object, is_scalar=True)).cast(dtype)
 
-    try:
+    if isinstance(value, np.generic):
         # numpy literals like np.float32(0) have item/dtype
         item = value.item()
 
@@ -176,7 +176,7 @@ def lit(
                 time_unit = dtype_name[len("timedelta64[") : -1]
                 return lit(item).cast(Duration(time_unit))
 
-    except AttributeError:
+    else:
         item = value
 
     return wrap_expr(plr.lit(item, allow_object, is_scalar=True))