fix: Specify waveform types (#155)

amazon-braket · Jul 19, 2023 · e8c5694 · e8c5694
1 parent cc8cc7e
commit e8c5694
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/braket/device_schema/pulse/native_gate_calibrations_v1.py b/src/braket/device_schema/pulse/native_gate_calibrations_v1.py
@@ -189,4 +189,4 @@ class NativeGateCalibrations(BraketSchemaBase):
     )
     braketSchemaHeader: BraketSchemaHeader = Field(default=_PROGRAM_HEADER, const=_PROGRAM_HEADER)
     gates: Dict[str, Dict[str, List[NativeGate]]]
-    waveforms: Dict[str, Waveform]
+    waveforms: Dict[str, Union[TemplateWaveform, ArbitraryWaveform]]