diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py
index e97e188..5dbba56 100644
--- a/ml_pytorch_mlflow.py
+++ b/ml_pytorch_mlflow.py
@@ -142,7 +142,7 @@ def my_main(epochs):
     
     # inputss, targetss = val_ds
     inputss = pd.DataFrame(inputss, dtype=np.float64)
-    # inputss = inputss.to_numpy()
+    inputss = inputss.to_numpy()
 
     MSE = mean_squared_error(expected, predicted)
     MAE = mean_absolute_error(expected, predicted)