Fixed missing hints after detuning and rabi (#632)

QuEraComputing · Sep 25, 2023 · 057db7e · 057db7e
1 parent 4c72a23
commit 057db7e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/bloqade/builder/coupling.py b/src/bloqade/builder/coupling.py
@@ -1,9 +1,10 @@
 from bloqade.builder.base import Builder
+from bloqade.builder.field import Rabi, Detuning
 
 
 class LevelCoupling(Builder):
     @property
-    def detuning(self):
+    def detuning(self) -> Detuning:
         """
         - Specify the Detuning field
         - Next-step: <SpacialModulation>
@@ -19,12 +20,11 @@ def detuning(self):
                 :: Address atom at location labeled by variable
 
         """
-        from bloqade.builder.field import Detuning
 
         return Detuning(self)
 
     @property
-    def rabi(self):
+    def rabi(self) -> Rabi:
         """
         - Specify the Rabi term/field.
         - Possible Next:
@@ -37,7 +37,6 @@ def rabi(self):
 
 
         """
-        from bloqade.builder.field import Rabi
 
         return Rabi(self)