@@ -152,11 +152,16 @@ def get_valid_nodes_in_range(self, model: ast.AST, range: Range, also_return: bo
152
152
)
153
153
154
154
if get_robot_version () >= (5 , 0 , 0 ):
155
- from robot .parsing .model .statements import Break , Continue , ReturnStatement , WhileHeader
156
-
157
- if get_robot_version () >= (6 , 0 , 0 ):
158
155
from robot .parsing .model .blocks import Try
159
- from robot .parsing .model .statements import ExceptHeader , FinallyHeader , TryHeader
156
+ from robot .parsing .model .statements import (
157
+ Break ,
158
+ Continue ,
159
+ ExceptHeader ,
160
+ FinallyHeader ,
161
+ ReturnStatement ,
162
+ TryHeader ,
163
+ WhileHeader ,
164
+ )
160
165
161
166
if not isinstance (model , (Keyword , TestCase )):
162
167
return []
@@ -232,9 +237,7 @@ def get_valid_nodes_in_range(self, model: ast.AST, range: Range, also_return: bo
232
237
),
233
238
)
234
239
or get_robot_version () >= (5 , 0 )
235
- and isinstance (n , WhileHeader )
236
- or get_robot_version () >= (6 , 0 )
237
- and isinstance (n , (TryHeader , ExceptHeader , FinallyHeader ))
240
+ and isinstance (n , (WhileHeader , TryHeader , ExceptHeader , FinallyHeader ))
238
241
):
239
242
return []
240
243
0 commit comments