Update src/util/smiles_cor.py

src/util/smiles_cor.py

@@ -1278,7 +1278,9 @@ class smi_correct(object):
                                         SRC)
         
         df_output["SMILES"] = df_output.apply(lambda row: self.standardization_pipeline(row["CORRECT"]), axis=1)
-        
-        df_output = self.remove_smiles_duplicates(df_output, subset="SMILES").drop(columns=["CORRECT", "INCORRECT", "ORIGINAL"]).dropna()
+
+        cols_to_drop = ["CORRECT", "INCORRECT", "ORIGINAL"]
+        cols_present = [col for col in cols_to_drop if col in df_output.columns]
+        df_output = self.remove_smiles_duplicates(df_output, subset="SMILES").drop(columns=cols_present).dropna()
         
         return df_output