21 lines
553 B
Python
21 lines
553 B
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
from ..core.celery_app import celery_app
|
||
|
|
from ..services.atp_model_service import execute_model_run_job
|
||
|
|
|
||
|
|
|
||
|
|
@celery_app.task(name="app.tasks.atp_model_tasks.execute_atp_model_run_job")
|
||
|
|
def execute_atp_model_run_job(
|
||
|
|
run_id: str,
|
||
|
|
payload_data: dict[str, Any],
|
||
|
|
actor_user_id: str | None,
|
||
|
|
) -> dict[str, str]:
|
||
|
|
execute_model_run_job(
|
||
|
|
run_id=run_id,
|
||
|
|
payload_data=payload_data,
|
||
|
|
actor_user_id=actor_user_id,
|
||
|
|
)
|
||
|
|
return {"run_id": run_id, "status": "done"}
|