Update utils.py

utils.py

@@ -13,7 +13,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib.lines import Line2D
 import torch 
-import wandb
+#import wandb
 RDLogger.DisableLog('rdApp.*')
 import warnings
 from multiprocessing import Pool
@@ -232,9 +232,9 @@ def logging(log_path, start_time, i, idx, loss, save_path, drug_smiles, edge, no
     loss.update({'MaxLen': maxlen})
     loss.update({'Atom_types': atom_types_average})
 
-    wandb.log({"Validity": valid, "Uniqueness": unique, "Novelty": novel_starting_mol,
-                "Novelty_akt": novel_akt, "SNN_chembl": snn_chembl, "SNN_akt": snn_akt,
-                  "MaxLen": maxlen, "Atom_types": atom_types_average})
+    #wandb.log({"Validity": valid, "Uniqueness": unique, "Novelty": novel_starting_mol,
+    #            "Novelty_akt": novel_akt, "SNN_chembl": snn_chembl, "SNN_akt": snn_akt,
+    #              "MaxLen": maxlen, "Atom_types": atom_types_average})
 
     for tag, value in loss.items():
         log += ", {}: {:.4f}".format(tag, value)