Merge pull request #578 from pfackeldey/pfackeldey/fix_new_known_scalar_dtype

fix: dtype inference in new_known_scalar

Author: martindurant

src/dask_awkward/lib/core.py

@@ -706,6 +706,8 @@ def new_known_scalar(
             dtype = np.dtype(int)
         elif isinstance(s, (float, np.floating)):
             dtype = np.dtype(float)
+        elif hasattr(s, "dtype"):
+            dtype = getattr(s, "dtype")
         else:
             dtype = np.dtype(type(s))
     else:

tests/test_core.py

@@ -466,6 +466,14 @@ def test_new_known_scalar() -> None:
     assert c.compute() == s1
 
 
+def test_new_known_scalar_from_array() -> None:
+    s = np.array(0.0)
+    c = new_known_scalar(s)
+    assert c.compute() == s
+    assert c._meta is not None
+    assert c.dtype == s.dtype
+
+
 def test_scalar_dtype() -> None:
     s = 2
     c = new_known_scalar(s)