Improving the robustness of the 'remove_columns' method

sdgx/data_processors/base.py

@@ -1,9 +1,9 @@
 from __future__ import annotations
 
 from typing import Any, Dict
-
 import pandas as pd
 
+from sdgx.log import logger
 from sdgx.data_models.metadata import Metadata
 from sdgx.exceptions import SynthesizerProcessorError
 
@@ -72,7 +72,10 @@ def remove_columns(tabular_data: pd.DataFrame, column_name_to_remove: list) -> p
         result_data = tabular_data.copy()
 
         # Remove specified columns
-        result_data = result_data.drop(columns=column_name_to_remove)
+        try:
+            result_data = result_data.drop(columns=column_name_to_remove)
+        except KeyError:
+            logger.warning('Duplicate column removal occurred, which might lead to unintended consequences.')
 
         return result_data