burtenshaw commited on
Commit
8fecf33
·
1 Parent(s): 45e7686

use username split

Browse files
Files changed (1) hide show
  1. criteria.py +1 -4
criteria.py CHANGED
@@ -37,10 +37,7 @@ def get_user_results(username: str) -> List[Dict]:
37
  List of user's quiz results
38
  """
39
  try:
40
- ds = load_dataset(EXAM_DATASET_ID, split="train")
41
-
42
- # Filter for this user's results
43
- user_results = ds.filter(lambda x: x["username"] == username)
44
 
45
  results = user_results.to_list()
46
  print(f"Found {len(results)} results for user {username}")
 
37
  List of user's quiz results
38
  """
39
  try:
40
+ user_results = load_dataset(path=EXAM_DATASET_ID, split=username)
 
 
 
41
 
42
  results = user_results.to_list()
43
  print(f"Found {len(results)} results for user {username}")