Eachan Johnson commited on
Commit
ec633d9
·
1 Parent(s): e1979f9

FIx progress bars

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -197,8 +197,7 @@ def predict_one(
197
  input_representation: str = 'smiles',
198
  predict: Union[Iterable[str], str] = 'smiles',
199
  extra_metrics: Optional[Union[Iterable[str], str]] = None,
200
- return_pd: bool = False,
201
- progress = gr.Progress(track_tqdm=True)
202
  ):
203
  prediction_df = convert_one(
204
  strings=strings,
@@ -269,8 +268,7 @@ def predict_file(
269
  predict: str = 'smiles',
270
  predict2: Optional[str] = None,
271
  extra_metrics: Optional[Union[Iterable[str], str]] = None,
272
- return_pd: bool = False,
273
- progress = gr.Progress(track_tqdm=True)
274
  ):
275
  predict = cast(predict, to=list)
276
  if predict2 is not None and predict2 in MODELBOXES:
@@ -431,7 +429,8 @@ def _predict_then_draw_then_download(
431
  predict: Union[Iterable[str], str] = 'smiles',
432
  extra_metrics: Optional[Union[Iterable[str], str]] = None,
433
  smiles_col: str = "smiles",
434
- legends: Optional[Union[str, Iterable[str]]] = None
 
435
  ):
436
  df, gr_df = predict_one(
437
  strings=strings,
@@ -453,7 +452,8 @@ def _load_then_predict_then_download_then_reveal_plot(
453
  input_representation: str = 'smiles',
454
  predict: str = 'smiles',
455
  predict2: Optional[str] = "",
456
- extra_metrics: Optional[Union[Iterable[str], str]] = None
 
457
  ):
458
  (df, df_gr), col_opts = load_input_data(
459
  file,
 
197
  input_representation: str = 'smiles',
198
  predict: Union[Iterable[str], str] = 'smiles',
199
  extra_metrics: Optional[Union[Iterable[str], str]] = None,
200
+ return_pd: bool = False
 
201
  ):
202
  prediction_df = convert_one(
203
  strings=strings,
 
268
  predict: str = 'smiles',
269
  predict2: Optional[str] = None,
270
  extra_metrics: Optional[Union[Iterable[str], str]] = None,
271
+ return_pd: bool = False
 
272
  ):
273
  predict = cast(predict, to=list)
274
  if predict2 is not None and predict2 in MODELBOXES:
 
429
  predict: Union[Iterable[str], str] = 'smiles',
430
  extra_metrics: Optional[Union[Iterable[str], str]] = None,
431
  smiles_col: str = "smiles",
432
+ legends: Optional[Union[str, Iterable[str]]] = None,
433
+ progress = gr.Progress(track_tqdm=True)
434
  ):
435
  df, gr_df = predict_one(
436
  strings=strings,
 
452
  input_representation: str = 'smiles',
453
  predict: str = 'smiles',
454
  predict2: Optional[str] = "",
455
+ extra_metrics: Optional[Union[Iterable[str], str]] = None,
456
+ progress = gr.Progress(track_tqdm=True)
457
  ):
458
  (df, df_gr), col_opts = load_input_data(
459
  file,