Sonja Topf commited on
Commit
a14921b
·
1 Parent(s): 338b5f6

removed imports

Browse files
Files changed (1) hide show
  1. predict.py +0 -4
predict.py CHANGED
@@ -1,9 +1,6 @@
1
  import torch
2
  import csv
3
  import subprocess
4
- import pandas as pd
5
- import logging
6
- import json
7
 
8
  from src.preprocess import create_clean_smiles
9
 
@@ -45,7 +42,6 @@ def predict(smiles_list):
45
  writer.writerow(["smiles"]) # header
46
  for smi in clean_smiles:
47
  writer.writerow([smi])
48
- logging.info("here")
49
  # predict
50
  command = [
51
  "chemprop", "predict",
 
1
  import torch
2
  import csv
3
  import subprocess
 
 
 
4
 
5
  from src.preprocess import create_clean_smiles
6
 
 
42
  writer.writerow(["smiles"]) # header
43
  for smi in clean_smiles:
44
  writer.writerow([smi])
 
45
  # predict
46
  command = [
47
  "chemprop", "predict",