diff --git a/model_data/simulation_system/training.py b/model_data/simulation_system/training.py
index 2a01485c..b37e7154 100644
--- a/model_data/simulation_system/training.py
+++ b/model_data/simulation_system/training.py
@@ -145,7 +145,10 @@ def training(
     plt.tight_layout()
     RESIDUAL_FILE = "residuals.png"
     plt.savefig(output_base / METRICS_FOLDER / RESIDUAL_FILE, dpi=120) 
-    plt.savefig(RESIDUAL_FILE, dpi=120) 
+
+    # TODO: for cml, we might want to have class that outputs all data and plots to add to the report
+    # If we want residual plot/ any plots, we will need to self host
+    # plt.savefig(RESIDUAL_FILE, dpi=120) 
 
     
     # TODO: introduce a seperate script for model optimisation, and from there, optimise for deployment