ardaatahan commited on
Commit
99e6f55
1 Parent(s): 88f77e2

device filters by exact match

Browse files
Files changed (1) hide show
  1. main.py +6 -11
main.py CHANGED
@@ -268,17 +268,11 @@ def performance_filter(
268
  ]
269
 
270
  # Filter by devices
271
- filtered_df = (
272
- filtered_df[
273
- (
274
- filtered_df["Device"].str.contains(
275
- "|".join(re.escape(q.strip()) for q in devices), case=False
276
- )
277
- )
278
- ]
279
- if devices
280
- else pd.DataFrame(columns=filtered_df.columns)
281
- )
282
 
283
  # Filter by operating systems
284
  filtered_df = (
@@ -293,6 +287,7 @@ def performance_filter(
293
  else pd.DataFrame(columns=filtered_df.columns)
294
  )
295
 
 
296
  # Apply short-form and long-form speed and tokens per second filters
297
  min_short_speed, max_short_speed = short_speed_slider
298
  min_long_speed, max_long_speed = long_speed_slider
 
268
  ]
269
 
270
  # Filter by devices
271
+ if devices:
272
+ filtered_df = filtered_df[filtered_df["Device"].isin(devices)]
273
+ else:
274
+ filtered_df = pd.DataFrame(columns=filtered_df.columns)
275
+
 
 
 
 
 
 
276
 
277
  # Filter by operating systems
278
  filtered_df = (
 
287
  else pd.DataFrame(columns=filtered_df.columns)
288
  )
289
 
290
+
291
  # Apply short-form and long-form speed and tokens per second filters
292
  min_short_speed, max_short_speed = short_speed_slider
293
  min_long_speed, max_long_speed = long_speed_slider