fix(raytune_learner): fix imports in raytune_learner

mathysgrapotte · Jan 29, 2025 · 5bf0481 · 5bf0481
1 parent 5e6faa0
commit 5bf0481
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/stimulus/learner/raytune_learner.py b/src/stimulus/learner/raytune_learner.py
@@ -15,7 +15,7 @@
 from torch import nn, optim
 from torch.utils.data import DataLoader, Dataset
 
-from stimulus.data.experiments import EncoderLoader
+from stimulus.data.loaders import EncoderLoader
 from stimulus.data.handlertorch import TorchDataset
 from stimulus.learner.predict import PredictWrapper
 from stimulus.utils.generic_utils import set_general_seeds