Eachan Johnson commited on
Commit
ac04716
·
1 Parent(s): 53ffacc

Fix file prediction

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -182,6 +182,7 @@ def predict_file(
182
  df: pd.DataFrame,
183
  column: str = 'smiles',
184
  input_representation: str = 'smiles',
 
185
  extra_metrics: Optional[Union[Iterable[str], str]] = None
186
  ):
187
  if extra_metrics is None:
 
182
  df: pd.DataFrame,
183
  column: str = 'smiles',
184
  input_representation: str = 'smiles',
185
+ predict: Union[Iterable[str], str] = 'smiles',
186
  extra_metrics: Optional[Union[Iterable[str], str]] = None
187
  ):
188
  if extra_metrics is None: