fix for when passing dtypes instead of dtype names

ActivitySim · Nov 24, 2024 · e408414 · e408414
1 parent ea8174f
commit e408414
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sharrow/relationships.py b/sharrow/relationships.py
@@ -931,6 +931,10 @@ def get_expr(
         -------
         DataArray
         """
+        if np.issubdtype(dtype, np.number):
+            dtype = dtype.__name__
+        elif dtype is bool:
+            dtype = "bool"
         try:
             if allow_native:
                 result = self[expression]