Skip to content

Commit 289fb1c

Browse files
committed
Serialize skipped stages
1 parent fc5f4b5 commit 289fb1c

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

autonima/models/types.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -296,6 +296,11 @@ def serialize_screening_dict(
296296
screening_dict: Dict[str, Any]
297297
) -> Dict[str, Any]:
298298
"""Serialize a screening dict, handling CriteriaMapping objects."""
299+
# Check if this stage is skipped
300+
if screening_dict.get('skip_stage', False):
301+
return {"skip_stage": True}
302+
303+
# Otherwise, serialize all fields
299304
result = {}
300305
for key, value in screening_dict.items():
301306
if isinstance(value, CriteriaMapping):

0 commit comments

Comments
 (0)