From ae0ff6c728f6d2bdd280cda991ef0ea26e3d69df Mon Sep 17 00:00:00 2001 From: sfluegel Date: Wed, 16 Jul 2025 11:10:51 +0200 Subject: [PATCH] fix import for XYBaseDataModule --- chebai/cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chebai/cli.py b/chebai/cli.py index 41742596..96262447 100644 --- 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