MohamedRashad commited on
Commit
89f161d
·
1 Parent(s): cbdd9d6

Add task type filtering in load_requests and remove debug print statement

Browse files
Files changed (1) hide show
  1. utils.py +1 -2
utils.py CHANGED
@@ -203,10 +203,9 @@ def load_requests(status_folder, task_type=None):
203
  continue # Skip files that can't be loaded
204
 
205
  df = pd.DataFrame(requests_data)
206
- print(df)
207
 
208
  # Filter by task type
209
- if task_type:
210
  df = df[df['task'] == task_type]
211
 
212
  return df
 
203
  continue # Skip files that can't be loaded
204
 
205
  df = pd.DataFrame(requests_data)
 
206
 
207
  # Filter by task type
208
+ if task_type and not df.empty:
209
  df = df[df['task'] == task_type]
210
 
211
  return df