burtenshaw commited on
Commit
2d32551
·
1 Parent(s): e28fc42

use sanitized username

Browse files
Files changed (1) hide show
  1. criteria.py +3 -2
criteria.py CHANGED
@@ -36,13 +36,14 @@ def get_user_results(username: str) -> List[Dict]:
36
  List of user's quiz results
37
  """
38
  try:
 
39
  user_results = load_dataset(
40
  path=EXAM_DATASET_ID,
41
  streaming=True,
42
- data_files=f"data/{username}-00000-of-00001.parquet",
43
  )
44
  results = list(user_results["train"])
45
- print(f"Found {len(results)} results for user {username}")
46
  return results
47
 
48
  except Exception as e:
 
36
  List of user's quiz results
37
  """
38
  try:
39
+ sanitized_username = username.replace("-", "000")
40
  user_results = load_dataset(
41
  path=EXAM_DATASET_ID,
42
  streaming=True,
43
+ data_files=f"data/{sanitized_username}-00000-of-00001.parquet",
44
  )
45
  results = list(user_results["train"])
46
+ print(f"Found {len(results)} results for user {sanitized_username}")
47
  return results
48
 
49
  except Exception as e: