Spaces:
Runtime error
Runtime error
Add comment
Browse files- run_evaluation_jobs.py +6 -1
run_evaluation_jobs.py
CHANGED
@@ -28,6 +28,7 @@ def main():
|
|
28 |
# Filter IDs for appropriate AutoTrain env (staging vs prod)
|
29 |
projects_df = projects_df.copy().query(f"autotrain_env == '{AUTOTRAIN_ENV}'")
|
30 |
projects_to_approve = projects_df["project_id"].astype(int).tolist()
|
|
|
31 |
print(f"π Found {len(projects_to_approve)} evaluation projects to approve!")
|
32 |
|
33 |
for project_id in projects_to_approve:
|
@@ -37,7 +38,7 @@ def main():
|
|
37 |
path=f"/projects/{project_id}",
|
38 |
token=HF_TOKEN,
|
39 |
domain=AUTOTRAIN_BACKEND_API,
|
40 |
-
).json()
|
41 |
print(project_info)
|
42 |
# Only start evaluation for projects with completed data processing (status=3)
|
43 |
if project_info["status"] == 3 and project_info["training_status"] == "not_started":
|
@@ -52,8 +53,12 @@ def main():
|
|
52 |
except Exception as e:
|
53 |
print(f"There was a problem obtaining the project info for project ID {project_id}")
|
54 |
print(f"Error message: {e}")
|
|
|
55 |
pass
|
56 |
|
|
|
|
|
|
|
57 |
|
58 |
if __name__ == "__main__":
|
59 |
typer.run(main)
|
|
|
28 |
# Filter IDs for appropriate AutoTrain env (staging vs prod)
|
29 |
projects_df = projects_df.copy().query(f"autotrain_env == '{AUTOTRAIN_ENV}'")
|
30 |
projects_to_approve = projects_df["project_id"].astype(int).tolist()
|
31 |
+
failed_approvals = []
|
32 |
print(f"π Found {len(projects_to_approve)} evaluation projects to approve!")
|
33 |
|
34 |
for project_id in projects_to_approve:
|
|
|
38 |
path=f"/projects/{project_id}",
|
39 |
token=HF_TOKEN,
|
40 |
domain=AUTOTRAIN_BACKEND_API,
|
41 |
+
).json()q
|
42 |
print(project_info)
|
43 |
# Only start evaluation for projects with completed data processing (status=3)
|
44 |
if project_info["status"] == 3 and project_info["training_status"] == "not_started":
|
|
|
53 |
except Exception as e:
|
54 |
print(f"There was a problem obtaining the project info for project ID {project_id}")
|
55 |
print(f"Error message: {e}")
|
56 |
+
failed_approvals.append(project_id)
|
57 |
pass
|
58 |
|
59 |
+
if len(failed_approvals) > 0:
|
60 |
+
print(f"π¨ Failed to approve {len(failed_approvals)} projects: {failed_approvals}")
|
61 |
+
|
62 |
|
63 |
if __name__ == "__main__":
|
64 |
typer.run(main)
|