diff --git a/src/train.py b/src/train.py
index a105c74..150ce5c 100644
--- a/src/train.py
+++ b/src/train.py
@@ -372,7 +372,6 @@ def predict(self, dataset, indices, return_targets=False):
             contributions[i, :] = (
                 self.standardizer.restore_cont(conts_std).numpy().flatten()
             )
-            # contributions[i, :] = pred_dict["contributions"].cpu().detach().numpy().flatten()
 
         predictions_dict = {
             "targets": targets,