diff --git a/src/run_modules/model_setup.py b/src/run_modules/model_setup.py
index 0563fc88948c7281f389ec300b294116abd7a491..e8259b2847ea4ede1b365f49778f019c004fa7f1 100644
--- a/src/run_modules/model_setup.py
+++ b/src/run_modules/model_setup.py
@@ -60,10 +60,11 @@ class ModelSetup(RunEnvironment):
         self.data_store.set("channels", channels, self.scope)
 
     def compile_model(self):
-        # optimizer = self.data_store.get("optimizer", self.scope)
-        # loss = self.model.loss
+        """
+        Compiles the keras model. Compile options are mandetory and have to be set by implementing set_compile() method
+        in child class of AbstractModelClass.
+        """
         compile_options = self.model.compile_options
-        # self.model.compile(optimizer=optimizer, loss=loss, **compile_options)
         self.model.compile(**compile_options)
         self.data_store.set("model", self.model, self.scope)