small fixes

qua-platform · Aug 23, 2024 · e632165 · e632165
1 parent 7d7b6f3
commit e632165
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 1 deletion.
diff --git a/quam/components/channels.py b/quam/components/channels.py
@@ -55,6 +55,7 @@
 
 
 __all__ = [
+    "Channel",
     "DigitalOutputChannel",
     "StickyChannelAddon",
     "SingleChannel",
@@ -165,6 +166,7 @@ class StickyChannelAddon(QuamComponent):
         digital (bool, optional): If False, the sticky parameters are not applied to
             digital outputs. Default is True.
     """
+
     duration: int
     enabled: bool = True
     analog: bool = True

diff --git a/tests/components/channels/test_in_IQ_out_single_channel.py b/tests/components/channels/test_in_IQ_out_single_channel.py
@@ -12,6 +12,7 @@ def test_in_single_channel_attr_annotations():
         "opx_input_Q",
     }
     assert set(attr_annotations["optional"]) == {
+        "sticky",
         "operations",
         "filter_fir_taps",
         "filter_iir_taps",
@@ -82,7 +83,7 @@ def test_generate_config_ports(qua_config):
             },
             "analog_outputs": {1: {"delay": 0, "shareable": False}},
         }
-    }    
+    }
 
     assert qua_config["elements"] == {
         "in_out_channel": {

diff --git a/tests/components/channels/test_in_single_channel.py b/tests/components/channels/test_in_single_channel.py
@@ -8,6 +8,7 @@ def test_in_single_channel_attr_annotations():
     assert set(attr_annotations["required"]) == {"opx_input"}
     assert set(attr_annotations["optional"]) == {
         "operations",
+        "sticky",
         "id",
         "digital_outputs",
         "opx_input_offset",

diff --git a/tests/components/channels/test_in_single_out_IQ_channel.py b/tests/components/channels/test_in_single_out_IQ_channel.py
@@ -14,6 +14,7 @@ def test_in_single_channel_attr_annotations():
     }
     assert set(attr_annotations["optional"]) == {
         "operations",
+        "sticky",
         "intermediate_frequency",
         "LO_frequency",
         "RF_frequency",