Skip to content

Commit e67408d

Browse files
committed
fix:import
1 parent aeb4b80 commit e67408d

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

scheduler/management/commands/import.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,14 +21,13 @@ def job_model_str(model_str: str) -> str:
2121

2222
def get_task_type(model_str: str) -> TaskType:
2323
model_str = job_model_str(model_str)
24-
if model_str not in MODEL_NAMES:
25-
raise ValueError(f"Invalid model {model_str}")
2624
if model_str == "CronTask":
2725
return TaskType.CRON
2826
elif model_str == "RepeatableTask":
2927
return TaskType.REPEATABLE
30-
elif model_str == "ScheduledTask":
28+
elif model_str in {"ScheduledTask", "OnceTask"}:
3129
return TaskType.ONCE
30+
raise ValueError(f"Invalid model {model_str}")
3231

3332

3433
def create_task_from_dict(task_dict: Dict[str, Any], update):

0 commit comments

Comments
 (0)