Fix row header

LQDJudge · Feb 14, 2025 · 1060056 · 1060056
1 parent 4d4b282
commit 1060056
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/judge/management/commands/generate_data.py b/judge/management/commands/generate_data.py
@@ -45,7 +45,7 @@ def gen_submissions():
     with open(os.path.join(settings.ML_DATA_PATH, "submissions.csv"), "w") as csvfile:
         f = csv.writer(csvfile)
         # Write headers
-        f.writerow(["user_id", "problem_id"])
+        f.writerow(["uid", "pid"])
 
         # Write rows from the dictionary
         for (user_id, problem_id), submission in submissions_dict.items():