pass inputs with PulseSequence

amazon-braket · Dec 19, 2023 · ab492d9 · ab492d9
1 parent 1bc972b
commit ab492d9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/braket/aws/aws_quantum_task.py b/src/braket/aws/aws_quantum_task.py
@@ -569,7 +569,12 @@ def _(
     *args,
     **kwargs,
 ) -> AwsQuantumTask:
-    create_task_kwargs.update({"action": OpenQASMProgram(source=pulse_sequence.to_ir()).json()})
+    openqasm_program = OpenQASMProgram(
+        source=pulse_sequence.to_ir(),
+        inputs=inputs if inputs else None,
+    )
+
+    create_task_kwargs.update({"action": openqasm_program.json()})
     task_arn = aws_session.create_quantum_task(**create_task_kwargs)
     return AwsQuantumTask(task_arn, aws_session, *args, **kwargs)