diff --git a/src/mednet/config/data/montgomery_shenzhen/default.py b/src/mednet/config/data/montgomery_shenzhen/default.py
index bc6a1f94bbe12d77e3d31b8496b9a4795c44572d..106bb415262932cd4fcfa954322f6f8d7353f554 100644
--- a/src/mednet/config/data/montgomery_shenzhen/default.py
+++ b/src/mednet/config/data/montgomery_shenzhen/default.py
@@ -7,6 +7,6 @@ split).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("default.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen/fold_0.py b/src/mednet/config/data/montgomery_shenzhen/fold_0.py
index 31933cf4b67964110b79115062d239c507a3ff75..d03c9c544afd5ec5a3abec3c17984d5e571c3bec 100644
--- a/src/mednet/config/data/montgomery_shenzhen/fold_0.py
+++ b/src/mednet/config/data/montgomery_shenzhen/fold_0.py
@@ -7,6 +7,6 @@ validation fold 0).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("fold-0.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen/fold_1.py b/src/mednet/config/data/montgomery_shenzhen/fold_1.py
index 8f4b3705cea93c18c4d9594f05abb122079bfdf6..948a75b07de9fe76df372964fec4c147063fcc8d 100644
--- a/src/mednet/config/data/montgomery_shenzhen/fold_1.py
+++ b/src/mednet/config/data/montgomery_shenzhen/fold_1.py
@@ -7,6 +7,6 @@ validation fold 1).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("fold-1.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen/fold_2.py b/src/mednet/config/data/montgomery_shenzhen/fold_2.py
index 5be0fa6326633c584fa8e392e42d2accd8a65d1e..c627fa2253cf034a402135f8575ae99042487682 100644
--- a/src/mednet/config/data/montgomery_shenzhen/fold_2.py
+++ b/src/mednet/config/data/montgomery_shenzhen/fold_2.py
@@ -7,6 +7,6 @@ validation fold 2).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("fold-2.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen/fold_3.py b/src/mednet/config/data/montgomery_shenzhen/fold_3.py
index 4c82b55d84d22e876c2cbc7229acac61f236de24..9ba252f3114625f05b21cbbad1a420c5ea9dc834 100644
--- a/src/mednet/config/data/montgomery_shenzhen/fold_3.py
+++ b/src/mednet/config/data/montgomery_shenzhen/fold_3.py
@@ -7,6 +7,6 @@ validation fold 3).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("fold-3.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen/fold_4.py b/src/mednet/config/data/montgomery_shenzhen/fold_4.py
index a3514eb41a74bd05831a7cf87ce327c646f1bc9f..720a3adc225b864c104a2cb8bd77250fde45e7f3 100644
--- a/src/mednet/config/data/montgomery_shenzhen/fold_4.py
+++ b/src/mednet/config/data/montgomery_shenzhen/fold_4.py
@@ -7,6 +7,6 @@ validation fold 4).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("fold-4.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen/fold_5.py b/src/mednet/config/data/montgomery_shenzhen/fold_5.py
index c81d7788bfd0cc1da7b60f14a33282969f99604d..c784a4ea01bfeacb2140b7ffdff5d06a618865ab 100644
--- a/src/mednet/config/data/montgomery_shenzhen/fold_5.py
+++ b/src/mednet/config/data/montgomery_shenzhen/fold_5.py
@@ -7,6 +7,6 @@ validation fold 5).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("fold-5.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen/fold_6.py b/src/mednet/config/data/montgomery_shenzhen/fold_6.py
index 69e826a55a2343019c8a7b5fe1588e48a9b259cf..eaa4647240a73b2ae6d9b8b4e5ce7a76f2a6ad16 100644
--- a/src/mednet/config/data/montgomery_shenzhen/fold_6.py
+++ b/src/mednet/config/data/montgomery_shenzhen/fold_6.py
@@ -7,6 +7,6 @@ validation fold 6).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("fold-6.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen/fold_7.py b/src/mednet/config/data/montgomery_shenzhen/fold_7.py
index 6212724647b60e13d28110f50f658403dac4d4dd..7a6dd33b281e0c41b0823dacba70d443e19310d2 100644
--- a/src/mednet/config/data/montgomery_shenzhen/fold_7.py
+++ b/src/mednet/config/data/montgomery_shenzhen/fold_7.py
@@ -7,6 +7,6 @@ validation fold 7).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("fold-7.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen/fold_8.py b/src/mednet/config/data/montgomery_shenzhen/fold_8.py
index e6d610983fd82e0056665fbc8a5c716063eb71fd..8f295426865d2523caffac09e52f1d67fd138379 100644
--- a/src/mednet/config/data/montgomery_shenzhen/fold_8.py
+++ b/src/mednet/config/data/montgomery_shenzhen/fold_8.py
@@ -7,6 +7,6 @@ validation fold 8).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("fold-8.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen/fold_9.py b/src/mednet/config/data/montgomery_shenzhen/fold_9.py
index c59826b3cca43910fa0c53933e54090de9c31242..1ec041b8c56923b60eb744f744dfe579bc3fd434 100644
--- a/src/mednet/config/data/montgomery_shenzhen/fold_9.py
+++ b/src/mednet/config/data/montgomery_shenzhen/fold_9.py
@@ -7,6 +7,6 @@ validation fold 9).
See :py:class:`.montgomery_shenzhen.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen.datamodule import DataModule
datamodule = DataModule("fold-9.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/default.py b/src/mednet/config/data/montgomery_shenzhen_indian/default.py
index d4004d4fd27066e29826024c295f939b670a0d2a..a1ac8047aa6f526b68e5b4df03a87a55de98b140 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/default.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/default.py
@@ -7,6 +7,6 @@ See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical
details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("default.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/fold_0.py b/src/mednet/config/data/montgomery_shenzhen_indian/fold_0.py
index 439919667afcb8a58c7b1f39a4120eb19e177c43..aecbbd107ab4a4d8e1bbe15b318deb8f58fe1238 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/fold_0.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/fold_0.py
@@ -8,6 +8,6 @@ See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical
details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("fold-0.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/fold_1.py b/src/mednet/config/data/montgomery_shenzhen_indian/fold_1.py
index a400b2c57126160d3c367c483aef323d96ab0b95..bef0e4aee628a0a22f1b49a5052ee9124d6a2ad9 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/fold_1.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/fold_1.py
@@ -7,6 +7,6 @@
See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("fold-1.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/fold_2.py b/src/mednet/config/data/montgomery_shenzhen_indian/fold_2.py
index a3fd8dd0dfd29e99c3edadc8172948d27cd14684..4a75d3329902f0ca439bf17bafea950f5bc23de8 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/fold_2.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/fold_2.py
@@ -8,6 +8,6 @@ See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical
details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("fold-2.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/fold_3.py b/src/mednet/config/data/montgomery_shenzhen_indian/fold_3.py
index 671e10fb94b244bcc4150224ada5f8d176935885..a85428cb606030252413c841b2849b128a98713c 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/fold_3.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/fold_3.py
@@ -8,6 +8,6 @@ See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical
details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("fold-3.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/fold_4.py b/src/mednet/config/data/montgomery_shenzhen_indian/fold_4.py
index df22c4aac1cb48a9d284696e763b99e52134bdb4..857a5b622fbe5f814b8aec418fc631297632083d 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/fold_4.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/fold_4.py
@@ -8,6 +8,6 @@ See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical
details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("fold-4.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/fold_5.py b/src/mednet/config/data/montgomery_shenzhen_indian/fold_5.py
index e85044381e3c5154273ce119fe5197b97a1c5bf6..58ac206ca6b30af085b4c7d742757414267e934a 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/fold_5.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/fold_5.py
@@ -8,6 +8,6 @@ See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical
details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("fold-5.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/fold_6.py b/src/mednet/config/data/montgomery_shenzhen_indian/fold_6.py
index 962d4443e7160dacfe0a9480a36ddb27276307c9..db77fcc0fcb92446a7220c806fa63a6b7b22bf4b 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/fold_6.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/fold_6.py
@@ -8,6 +8,6 @@ See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical
details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("fold-6.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/fold_7.py b/src/mednet/config/data/montgomery_shenzhen_indian/fold_7.py
index bf08e64f18a92555afa97492f1a1b365c0a334b5..aa7f79e5d7d18d7676953ec208da8fc7acddbc54 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/fold_7.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/fold_7.py
@@ -7,6 +7,6 @@
See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("fold-7.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/fold_8.py b/src/mednet/config/data/montgomery_shenzhen_indian/fold_8.py
index dbc23e04a9f76f724fbb15e1525030208ede585a..31fca75024ac528cc04879fbddc3150367026239 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/fold_8.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/fold_8.py
@@ -8,6 +8,6 @@ See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical
details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("fold-8.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian/fold_9.py b/src/mednet/config/data/montgomery_shenzhen_indian/fold_9.py
index 885ffc9b25478737ba3fb9521bffbf6f01f8a25a..54c9a07e798db588cd2857331485302fd770af76 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian/fold_9.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian/fold_9.py
@@ -7,6 +7,6 @@
See :py:class:`.montgomery_shenzhen_indian.datamodule.DataModule` for technical details.
"""
-from mednet.data.config.montgomery_shenzhen_indian.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_indian.datamodule import DataModule
datamodule = DataModule("fold-9.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_padchest/default.py b/src/mednet/config/data/montgomery_shenzhen_indian_padchest/default.py
index 3922eac6abcbe47eb5b265306c0786e3fd6c01cd..d3f7c61b01162ffd1032c88b43c3e78eef837b6b 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_padchest/default.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_padchest/default.py
@@ -4,7 +4,7 @@
"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and Padchest
datasets."""
-from mednet.data.config.montgomery_shenzhen_padchest.datamodule import (
+from mednet.config.data.montgomery_shenzhen_padchest.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_0.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_0.py
index 6d301af045a24637cf6984035b6671c9276bf97b..d688e8edbcf0ab7d9c1158fa2dbf15cfbe99a8bd 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_0.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_0.py
@@ -8,6 +8,6 @@ This remix dataset combines ``fold-0`` from Montgomery, Shenzhen, and Indian
datasets with ``v1-fold-0`` of TBX11k (healthy vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_tbx11k.datamodule import DataModule
+from mednet.config.data.montgomery_shenzhen_tbx11k.datamodule import DataModule
datamodule = DataModule("fold-0.json", "v1-fold-0.json")
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_1.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_1.py
index f3476877d017eb9b03fa0f3a99d2ac94e33e8dd9..10f57c42bcc895ef47e447150859cbb7665f3d43 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_1.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_1.py
@@ -8,7 +8,7 @@ This remix dataset combines ``fold-1`` from Montgomery, Shenzhen, and Indian
datasets with ``v1-fold-1`` of TBX11k (healthy vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_2.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_2.py
index ba5e0e29cb261723d17ebdc1eddfe08b7f63d36b..6ec20140b059eb0f98c60341479d0ebefd318ca7 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_2.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_2.py
@@ -8,7 +8,7 @@ This remix dataset combines ``fold-2`` from Montgomery, Shenzhen, and Indian
datasets with ``v1-fold-2`` of TBX11k (healthy vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_3.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_3.py
index 1af83edf8689962b09ec4c50a75933f4ff21492e..569ed8017632595bd2627a46571587f6bfb0f898 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_3.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_3.py
@@ -8,7 +8,7 @@ This remix dataset combines ``fold-3`` from Montgomery, Shenzhen, and Indian
datasets with ``v1-fold-3`` of TBX11k (healthy vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_4.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_4.py
index 0b5aa4c0bc076a6f873517747199c54196c27b7d..04a74cd35ccb133634d8ca5db18936861ef2eab0 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_4.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_4.py
@@ -8,7 +8,7 @@ This remix dataset combines ``fold-4`` from Montgomery, Shenzhen, and Indian
datasets with ``v1-fold-4`` of TBX11k (healthy vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_5.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_5.py
index 946554e3a437f64eb78c3e08e55559659dd4f566..b528a7b6799f76da1cb8c50c8071f088f6c082e2 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_5.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_5.py
@@ -8,7 +8,7 @@ This remix dataset combines ``fold-5`` from Montgomery, Shenzhen, and Indian
datasets with ``v1-fold-5`` of TBX11k (healthy vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_6.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_6.py
index 29ba981daddf32f9bdd54e893ccc811b8e74b0ef..fdf55b4274f8329dda7aa28f654bde654df2fc33 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_6.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_6.py
@@ -8,7 +8,7 @@ This remix dataset combines ``fold-6`` from Montgomery, Shenzhen, and Indian
datasets with ``v1-fold-6`` of TBX11k (healthy vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_7.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_7.py
index a2cf115918c5a2af0492759b420e568279d329b1..833d2c04b2b8699e5143f9ad99231440133ff9bc 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_7.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_7.py
@@ -8,7 +8,7 @@ This remix dataset combines ``fold-7`` from Montgomery, Shenzhen, and Indian
datasets with ``v1-fold-7`` of TBX11k (healthy vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_8.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_8.py
index bd023982b29a5003e0155b44545b999bf3fd919c..a0dd7e822386ea260808a0ead9475d78136ce91e 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_8.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_8.py
@@ -8,7 +8,7 @@ This remix dataset combines ``fold-8`` from Montgomery, Shenzhen, and Indian
datasets with ``v1-fold-8`` of TBX11k (healthy vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_9.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_9.py
index 4c64d08944fdd0ee9146da1ed2c29d04fe886bd5..ea00d8ecf7726c308fc9007031ab7af21b5f4008 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_9.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_fold_9.py
@@ -8,7 +8,7 @@ This remix dataset combines ``fold-9`` from Montgomery, Shenzhen, and Indian
datasets with ``v1-fold-9`` of TBX11k (healthy vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_healthy_vs_atb.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_healthy_vs_atb.py
index 0ff7a0b44e80940e13a72e64c6834c75809ad704..278ac64eeadc31a36b5df90ee5c76ccdc92e7a09 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_healthy_vs_atb.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v1_healthy_vs_atb.py
@@ -9,7 +9,7 @@ and Indian datasets with ``v1-healthy-vs-atb`` split of TBX11k (healthy vs.
active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_0.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_0.py
index e71fb091a774a8abc36a5969c0d75751e854c29d..022824a1795eb56bd7e95ccf7845b86452020098 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_0.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_0.py
@@ -9,7 +9,7 @@ datasets with ``v2-fold-0`` of TBX11k (healthy, sick and latent TB vs. active
TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_1.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_1.py
index 0e79b6ea170ea372ec15a5c38cde9a7954604ae0..e48e83e797e6f69942d7345d387f5e50d2946c67 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_1.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_1.py
@@ -9,7 +9,7 @@ datasets with ``v2-fold-1`` of TBX11k (healthy, sick and latent TB vs. active
TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_2.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_2.py
index 02054b2807c06bbc3707a9fd4bf64fc9e216d160..1e8f38fbd000c8e026c5451abfceffe3338bbd00 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_2.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_2.py
@@ -9,7 +9,7 @@ datasets with ``v2-fold-2`` of TBX11k (healthy, sick and latent TB vs. active
TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_3.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_3.py
index 89ff7a55dcba7ac62515b64f3ea11cde934ec7ba..553ea54762ee64a672a02f9e7478ef2bd21b2405 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_3.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_3.py
@@ -9,7 +9,7 @@ datasets with ``v2-fold-3`` of TBX11k (healthy, sick and latent TB vs. active
TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_4.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_4.py
index 0b2fd6d115b0248c043a7263becb07ab89226bcc..672c9bbe3a09c88b6b50ac27d6dccbbf594a2e82 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_4.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_4.py
@@ -9,7 +9,7 @@ datasets with ``v2-fold-4`` of TBX11k (healthy, sick and latent TB vs. active
TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_5.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_5.py
index dc0d77023b3954faeb8bbd85d0fbee03695d8ec7..b7b3c5eddaa9daf3b1181b422667c48991d77628 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_5.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_5.py
@@ -9,7 +9,7 @@ datasets with ``v2-fold-5`` of TBX11k (healthy, sick and latent TB vs. active
TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_6.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_6.py
index dc2b3792f561d12294ca7101281026a0e99150f2..e30017374dcb6b78bb9b829e5956878666551f05 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_6.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_6.py
@@ -9,7 +9,7 @@ datasets with ``v2-fold-6`` of TBX11k (healthy, sick and latent TB vs. active
TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_7.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_7.py
index eceb342b29ea1622c8b6806763993cf6d4ec3924..4f3b3e9167c998306e4836840f51fded3de5963b 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_7.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_7.py
@@ -9,7 +9,7 @@ datasets with ``v2-fold-7`` of TBX11k (healthy, sick and latent TB vs. active
TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_8.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_8.py
index e4ff80593325b9e6ae834f0d8b96383aa1ac475d..afdb0f75625348df60c5b01760cd541d684c2dae 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_8.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_8.py
@@ -9,7 +9,7 @@ datasets with ``v2-fold-8`` of TBX11k (healthy, sick and latent TB vs. active
TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_9.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_9.py
index 1aeee35ecb2b0e4489423725056ada7d24b5d70b..31b1e4400fd22674df823791d404fc3d0817e347 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_9.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_fold_9.py
@@ -9,7 +9,7 @@ datasets with ``v2-fold-9`` of TBX11k (healthy, sick and latent TB vs. active
TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_others_vs_atb.py b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_others_vs_atb.py
index 82a53cbb084b00e0e8492aa095a86d6596d5cbf7..9dca260604e2c65e85f9a1b06cd96fd853560e3d 100644
--- a/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_others_vs_atb.py
+++ b/src/mednet/config/data/montgomery_shenzhen_indian_tbx11k/v2_others_vs_atb.py
@@ -9,7 +9,7 @@ and Indian datasets with ``v2-others-vs-atb`` of TBX11k (healthy, sick and
latent TB vs. active TB samples).
"""
-from mednet.data.config.montgomery_shenzhen_indian_tbx11k.datamodule import (
+from mednet.config.data.montgomery_shenzhen_indian_tbx11k.datamodule import (
DataModule,
)
diff --git a/src/mednet/config/data/nih_cxr14_padchest/idiap.py b/src/mednet/config/data/nih_cxr14_padchest/idiap.py
index 91d74d9b670f75607fb4daa32e0bb13c069d7f10..e7781517bce793fd1507c164ddfc8b6c0acb9758 100644
--- a/src/mednet/config/data/nih_cxr14_padchest/idiap.py
+++ b/src/mednet/config/data/nih_cxr14_padchest/idiap.py
@@ -4,6 +4,6 @@
"""Aggregated dataset composed of NIH CXR14 relabeld and PadChest (normalized)
datasets (no-tb-idiap split)."""
-from mednet.data.config.nih_cx14_padchest.datamodule import DataModule
+from mednet.config.data.nih_cx14_padchest.datamodule import DataModule
datamodule = DataModule("default.json.bz2", "no-tb-idiap.json.bz2")