Skip to content

Commit 057db7e

Browse files
authored
Fixed missing hints after detuning and rabi (#632)
1 parent 4c72a23 commit 057db7e

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

src/bloqade/builder/coupling.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
from bloqade.builder.base import Builder
2+
from bloqade.builder.field import Rabi, Detuning
23

34

45
class LevelCoupling(Builder):
56
@property
6-
def detuning(self):
7+
def detuning(self) -> Detuning:
78
"""
89
- Specify the Detuning field
910
- Next-step: <SpacialModulation>
@@ -19,12 +20,11 @@ def detuning(self):
1920
:: Address atom at location labeled by variable
2021
2122
"""
22-
from bloqade.builder.field import Detuning
2323

2424
return Detuning(self)
2525

2626
@property
27-
def rabi(self):
27+
def rabi(self) -> Rabi:
2828
"""
2929
- Specify the Rabi term/field.
3030
- Possible Next:
@@ -37,7 +37,6 @@ def rabi(self):
3737
3838
3939
"""
40-
from bloqade.builder.field import Rabi
4140

4241
return Rabi(self)
4342

0 commit comments

Comments
 (0)