angee.jobs.tasks
Celery task wrappers owned by the framework job seam.
rebalance_fractional_ranks
python
@shared_task(
name="jobs.rebalance_fractional_ranks",
autoretry_for=(Exception, ),
retry_backoff=True,
retry_kwargs={"max_retries": 3},
)
def rebalance_fractional_ranks(model_label: str, context: dict[str, Any],
rank_field: str) -> intRebalance one concrete model's ranks inside a Celery-serializable context.