From 53b96a3f3b83788fc2d7a16860212dfba35691e9 Mon Sep 17 00:00:00 2001
From: Julien Roussel <3178729-JulienRoussel77@users.noreply.gitlab.com>
Date: Sat, 30 Aug 2025 20:20:21 +0200
Subject: [PATCH] print removed

---
 qolmat/imputations/imputers_pytorch.py | 8 --------
 1 file changed, 8 deletions(-)

diff --git a/qolmat/imputations/imputers_pytorch.py b/qolmat/imputations/imputers_pytorch.py
index cad9a3ce..35e4644f 100644
--- a/qolmat/imputations/imputers_pytorch.py
+++ b/qolmat/imputations/imputers_pytorch.py
@@ -107,14 +107,6 @@ def _fit_estimator(
         optimizer = optim.Adam(estimator.parameters(), lr=self.learning_rate)
         loss_fn = self.loss_fn
 
-        # if X.shape[0] != estimator[0].in_features:
-        #     raise ValueError(
-        #         "The number of features in X does not match the input "
-        #         "features of the estimator. The estimator expects"
-        #         f" {estimator[0].in_features} features, but X has "
-        #         f"{X.shape[0]} features."
-        #     )
-
         with tqdm(total=self.epochs, desc="Training", unit="epoch") as pbar:
             for _ in range(self.epochs):
                 estimator.train()