jannisborn commited on
Commit
9db0a68
·
unverified ·
1 Parent(s): 7b49ccf
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -49,6 +49,7 @@ def draw_grid_predict(
49
  obj = mols2grid.display(
50
  result,
51
  tooltip=list(result.keys()),
 
52
  height=1100,
53
  n_cols=n_cols,
54
  name="Results",
 
49
  obj = mols2grid.display(
50
  result,
51
  tooltip=list(result.keys()),
52
+ subset=["img"] + list(result.keys()),
53
  height=1100,
54
  n_cols=n_cols,
55
  name="Results",