[960] debug unit tests 11

ToFuProject · Sep 6, 2024 · 1d153b4 · 1d153b4
1 parent 99428d4
commit 1d153b4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tofu/geom/_comp.py b/tofu/geom/_comp.py
@@ -594,7 +594,12 @@ def _Ves_get_sample_checkinputs(
         msg += "  You provided:\n{}".format(ind)
         raise Exception(msg)
 
-    return res, domain, resMode, ind.astype(np.int64)
+    if isinstance(ind, np.ndarray):
+        ind = ind.astype(np.int64)
+    elif isinstance(ind, list):
+        ind = [ii.astype(np.int64) for ii in ind]
+
+    return res, domain, resMode, ind
 
 
 def _Ves_get_sampleEdge(