Update inference.py

inference.py

@@ -223,10 +223,11 @@ class Inference(object):
         # post-process * to Carbon atom in valid molecules
 
         return{
-            "Validity": f"fraction_valid(metric_calc_dr):.2f",
-            "Uniqueness": f"fraction_unique(uniqueness_calc):.2f",
-            "Novelty (Train)": f"novelty(metric_calc_dr, chembl_smiles):.2f",
-            "Novelty (Inference)": f"novelty(metric_calc_dr, chembl_test):.2f",
+            "Runtime (seconds)": f"{et}:.2f",
+            "Validity": f"{fraction_valid(metric_calc_dr)}:.2f",
+            "Uniqueness": f"{fraction_unique(uniqueness_calc)}:.2f",
+            "Novelty (Train)": f"{novelty(metric_calc_dr, chembl_smiles)}:.2f",
+            "Novelty (Inference)": f"{novelty(metric_calc_dr, chembl_test)}:.2f",
         }