diff --git a/src/stimulus/learner/raytune_learner.py b/src/stimulus/learner/raytune_learner.py
index be2c2de..e147641 100644
--- a/src/stimulus/learner/raytune_learner.py
+++ b/src/stimulus/learner/raytune_learner.py
@@ -15,8 +15,8 @@
 from torch import nn, optim
 from torch.utils.data import DataLoader, Dataset
 
-from stimulus.data.loaders import EncoderLoader
 from stimulus.data.handlertorch import TorchDataset
+from stimulus.data.loaders import EncoderLoader
 from stimulus.learner.predict import PredictWrapper
 from stimulus.utils.generic_utils import set_general_seeds
 from stimulus.utils.yaml_model_schema import RayTuneModel