Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
89f161d
1
Parent(s):
cbdd9d6
Add task type filtering in load_requests and remove debug print statement
Browse files
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
|