diff --git a/fedot/core/operations/evaluation/operation_implementations/models/ts_implementations/cgru.py b/fedot/core/operations/evaluation/operation_implementations/models/ts_implementations/cgru.py
index 19b1f15ea8..d4ab69b86b 100644
--- a/fedot/core/operations/evaluation/operation_implementations/models/ts_implementations/cgru.py
+++ b/fedot/core/operations/evaluation/operation_implementations/models/ts_implementations/cgru.py
@@ -133,8 +133,7 @@ def _create_dataloader(self, input_data: InputData):
         :return torch.utils.data.DataLoader: DataLoader with train data
         """
         x, y = self._fit_transform_scaler(input_data)
-        x = torch.from_numpy(x).float()
-        y = torch.from_numpy(y).float()
+        x, y = (torch.from_numpy(np_data.astype(float)) for np_data in (x, y))
         return DataLoader(TensorDataset(x, y), batch_size=self.params.get("batch_size"))
 
     def _fit_transform_scaler(self, data: InputData):