fix

amazon-braket · Sep 6, 2024 · 8ecf2d2 · 8ecf2d2
1 parent 1c04013
commit 8ecf2d2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/braket/aws/aws_quantum_task.py b/src/braket/aws/aws_quantum_task.py
@@ -582,7 +582,7 @@ def _(
     aws_session: AwsSession,
     create_task_kwargs: dict[str, Any],
     device_arn: str,
-    device_parameters: dict | BraketSchemaBase,
+    device_parameters: Union[dict, BraketSchemaBase],  # noqa: UP007
     _disable_qubit_rewiring: bool,
     inputs: dict[str, float],
     gate_definitions: dict[tuple[Gate, QubitSet], PulseSequence],
@@ -619,7 +619,7 @@ def _(
     aws_session: AwsSession,
     create_task_kwargs: dict[str, Any],
     device_arn: str,
-    device_parameters: dict | BraketSchemaBase,
+    device_parameters: Union[dict, BraketSchemaBase],  # noqa: UP007
     _disable_qubit_rewiring: bool,
     inputs: dict[str, float],
     gate_definitions: Optional[dict[tuple[Gate, QubitSet], PulseSequence]],  # noqa: UP007
@@ -647,7 +647,7 @@ def _(
     aws_session: AwsSession,
     create_task_kwargs: dict[str, any],
     device_arn: str,
-    _device_parameters: dict | BraketSchemaBase,
+    _device_parameters: Union[dict, BraketSchemaBase],  # noqa: UP007
     _disable_qubit_rewiring: bool,
     inputs: dict[str, float],
     gate_definitions: dict[tuple[Gate, QubitSet], PulseSequence],