Merge pull request #107 from ChEB-AI/fix/cli-imports

sfluegel05 · web-flow · commit 713b7aebcd78 · 2025-07-18T17:50:28.000+02:00
fix import for XYBaseDataModule
diff --git a/chebai/cli.py b/chebai/cli.py
@@ -2,7 +2,7 @@
 
 from lightning.pytorch.cli import LightningArgumentParser, LightningCLI
 
-from chebai.preprocessing.datasets import XYBaseDataModule
+from chebai.preprocessing.datasets.base import XYBaseDataModule
 from chebai.trainer.CustomTrainer import CustomTrainer
 
 
