fix: batch tasking passing lists to single tasks (#917)

amazon-braket · Mar 21, 2024 · 314bf59 · 314bf59
1 parent 4e9ae75
commit 314bf59
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/braket/aws/aws_quantum_task_batch.py b/src/braket/aws/aws_quantum_task_batch.py
@@ -201,7 +201,7 @@ def _tasks_inputs_gatedefs(
                     batch_length = arg_length
 
         for i, arg_length in enumerate(arg_lengths):
-            if arg_length == 1:
+            if isinstance(args[i], (dict, single_task_type)):
                 args[i] = repeat(args[i], batch_length)
 
         tasks_inputs_definitions = list(zip(*args))