Check type inside Sequence and Iterators

unitaryfund · Oct 30, 2024 · 7c4a60f · 7c4a60f
1 parent 3c7b027
commit 7c4a60f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/mitiq/executor/executor.py b/mitiq/executor/executor.py
@@ -208,7 +208,12 @@ def evaluate(
         # check returned type
         manual_return_type = None
         if len(all_results) > 0:
-            manual_return_type = type(all_results[0])
+            if isinstance(all_results[0], Sequence):
+                manual_return_type = type(all_results[0][0])
+            elif isinstance(all_results[0], Iterable):
+                manual_return_type = type(next(iter(all_results[0])))
+            else:
+                manual_return_type = type(all_results[0])
 
         # Parse the results.
         if manual_return_type in FloatLike: