fix import from main

amazon-braket · Oct 7, 2024 · 89984d8 · 89984d8
1 parent 8e25245
commit 89984d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/braket/devices/local_simulator.py b/src/braket/devices/local_simulator.py
@@ -286,7 +286,7 @@ def _(self, program: OpenQASMProgram, inputs: Optional[dict[str, float]], _shots
         return program
 
     @_construct_payload.register
-    def _(self, program: SerializableProgram, inputs: Optional[dict[str, float]], _shots):
+    def _(self, program: SerializableProgram, inputs: dict[str, float] | None, _shots: int):
         inputs_copy = inputs.copy() if inputs is not None else {}
         return OpenQASMProgram(source=program.to_ir(ir_type=IRType.OPENQASM), inputs=inputs_copy)