Dun3Co commited on
Commit
d250715
·
verified ·
1 Parent(s): f87d4de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -163,8 +163,8 @@ def metrics(batch: Optional[BatchInputData] = None, y: Optional[List[int]] = Non
163
  X = fetch_test_data(limit=limit)
164
 
165
  if y is None:
166
- # Look for 'y_true' column in NoCoDB data
167
- if "y_" in X.columns:
168
  y = X["y"].astype(int).tolist()
169
  X = X.drop(columns=["y"])
170
  else:
 
163
  X = fetch_test_data(limit=limit)
164
 
165
  if y is None:
166
+ # Look for 'y' column in NoCoDB data
167
+ if "y" in X.columns:
168
  y = X["y"].astype(int).tolist()
169
  X = X.drop(columns=["y"])
170
  else: