diff --git a/api/crud.py b/api/crud.py index 474185f..ed9ae93 100755 --- a/api/crud.py +++ b/api/crud.py @@ -246,7 +246,7 @@ def update_experimentset( return db_expset -def remove_db_expset(db: Session, experimentset_id: int) -> bool: +def remove_experimentset(db: Session, experimentset_id: int) -> bool: db_expset = db.query(models.ExperimentSet).get(experimentset_id) if db_expset is None: return False @@ -354,7 +354,7 @@ def get_leaderboard(db: Session, metric_name: str = "judge_notator", dataset_nam .filter(and_(models.Result.experiment_id == result.experiment_id, models.Result.metric_name != metric_name))\ .all() - + entry = schemas.LeaderboardEntry( experiment_id=result.experiment_id, model_name=result.model_name, diff --git a/api/endpoints.py b/api/endpoints.py index b676952..0fe2e00 100755 --- a/api/endpoints.py +++ b/api/endpoints.py @@ -147,11 +147,14 @@ def patch_experiment( return db_exp -@router.delete("/experiment/{id}") +@router.delete( + "/experiment/{id}", + tags=["experiments"], +) def delete_experiment( id: int, db: Session = Depends(get_db), - tags=["experiments"], + admin_check=Depends(admin_only), ): if not crud.remove_experiment(db, id): raise HTTPException(status_code=404, detail="Experiment not found") @@ -335,11 +338,12 @@ def retry_runs(id: int, db: Session = Depends(get_db)): # LeaderBoard # + @router.get("/leaderboard", response_model=schemas.Leaderboard, tags=["leaderboard"]) def read_leaderboard( metric_name: str = "judge_notator", dataset_name: str = None, limit: int = 100, - db: Session = Depends(get_db) + db: Session = Depends(get_db), ): return crud.get_leaderboard(db, metric_name=metric_name, dataset_name=dataset_name, limit=limit)