From 8510560a445d6cffc0afc4e5a64a7e76c38bfe4f Mon Sep 17 00:00:00 2001
From: "ogueler@idiap.ch" <ogueler@vws110.idiap.ch>
Date: Sat, 15 Apr 2023 03:26:47 +0200
Subject: [PATCH] fixed formatting
---
.../configs/datasets/mc_ch_in_11k/__init__.py | 25 +-
.../configs/datasets/mc_ch_in_11k/default.py | 4 +-
.../configs/datasets/mc_ch_in_11k/fold_0.py | 4 +-
.../datasets/mc_ch_in_11k/fold_0_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k/fold_1.py | 4 +-
.../datasets/mc_ch_in_11k/fold_1_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k/fold_2.py | 4 +-
.../datasets/mc_ch_in_11k/fold_2_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k/fold_3.py | 4 +-
.../datasets/mc_ch_in_11k/fold_3_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k/fold_4.py | 4 +-
.../datasets/mc_ch_in_11k/fold_4_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k/fold_5.py | 4 +-
.../datasets/mc_ch_in_11k/fold_5_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k/fold_6.py | 4 +-
.../datasets/mc_ch_in_11k/fold_6_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k/fold_7.py | 4 +-
.../datasets/mc_ch_in_11k/fold_7_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k/fold_8.py | 4 +-
.../datasets/mc_ch_in_11k/fold_8_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k/fold_9.py | 4 +-
.../datasets/mc_ch_in_11k/fold_9_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k/rgb.py | 4 +-
.../datasets/mc_ch_in_11k_RS/__init__.py | 25 +-
.../datasets/mc_ch_in_11k_RS/default.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_0.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_0_rgb.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_1.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_1_rgb.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_2.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_2_rgb.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_3.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_3_rgb.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_4.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_4_rgb.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_5.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_5_rgb.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_6.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_6_rgb.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_7.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_7_rgb.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_8.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_8_rgb.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_9.py | 4 +-
.../datasets/mc_ch_in_11k_RS/fold_9_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11k_RS/rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2/__init__.py | 25 +-
.../datasets/mc_ch_in_11kv2/default.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/fold_0.py | 4 +-
.../datasets/mc_ch_in_11kv2/fold_0_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/fold_1.py | 4 +-
.../datasets/mc_ch_in_11kv2/fold_1_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/fold_2.py | 4 +-
.../datasets/mc_ch_in_11kv2/fold_2_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/fold_3.py | 4 +-
.../datasets/mc_ch_in_11kv2/fold_3_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/fold_4.py | 4 +-
.../datasets/mc_ch_in_11kv2/fold_4_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/fold_5.py | 4 +-
.../datasets/mc_ch_in_11kv2/fold_5_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/fold_6.py | 4 +-
.../datasets/mc_ch_in_11kv2/fold_6_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/fold_7.py | 4 +-
.../datasets/mc_ch_in_11kv2/fold_7_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/fold_8.py | 4 +-
.../datasets/mc_ch_in_11kv2/fold_8_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/fold_9.py | 4 +-
.../datasets/mc_ch_in_11kv2/fold_9_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2/rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/__init__.py | 25 +-
.../datasets/mc_ch_in_11kv2_RS/default.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_0.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_0_rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_1.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_1_rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_2.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_2_rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_3.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_3_rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_4.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_4_rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_5.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_5_rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_6.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_6_rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_7.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_7_rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_8.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_8_rgb.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_9.py | 4 +-
.../datasets/mc_ch_in_11kv2_RS/fold_9_rgb.py | 4 +-
.../configs/datasets/mc_ch_in_11kv2_RS/rgb.py | 4 +-
.../datasets/tbx11k_simplified_v2/rgb.py | 3 +-
.../datasets/tbx11k_simplified_v3/default.py | 2 +-
.../datasets/tbx11k_simplified_v3/fold_0.py | 2 +-
.../tbx11k_simplified_v3/fold_0_rgb.py | 2 +-
.../datasets/tbx11k_simplified_v3/fold_1.py | 2 +-
.../tbx11k_simplified_v3/fold_1_rgb.py | 2 +-
.../datasets/tbx11k_simplified_v3/fold_2.py | 2 +-
.../tbx11k_simplified_v3/fold_2_rgb.py | 2 +-
.../datasets/tbx11k_simplified_v3/fold_3.py | 2 +-
.../tbx11k_simplified_v3/fold_3_rgb.py | 2 +-
.../datasets/tbx11k_simplified_v3/fold_4.py | 2 +-
.../tbx11k_simplified_v3/fold_4_rgb.py | 2 +-
.../datasets/tbx11k_simplified_v3/fold_5.py | 2 +-
.../tbx11k_simplified_v3/fold_5_rgb.py | 2 +-
.../datasets/tbx11k_simplified_v3/fold_6.py | 2 +-
.../tbx11k_simplified_v3/fold_6_rgb.py | 2 +-
.../datasets/tbx11k_simplified_v3/fold_7.py | 2 +-
.../tbx11k_simplified_v3/fold_7_rgb.py | 2 +-
.../datasets/tbx11k_simplified_v3/fold_8.py | 2 +-
.../tbx11k_simplified_v3/fold_8_rgb.py | 2 +-
.../datasets/tbx11k_simplified_v3/fold_9.py | 2 +-
.../tbx11k_simplified_v3/fold_9_rgb.py | 2 +-
.../datasets/tbx11k_simplified_v3/rgb.py | 5 +-
.../tbx11k_simplified_v3_RS/default.py | 2 +-
.../tbx11k_simplified_v3_RS/fold_0.py | 2 +-
.../tbx11k_simplified_v3_RS/fold_1.py | 2 +-
.../tbx11k_simplified_v3_RS/fold_2.py | 2 +-
.../tbx11k_simplified_v3_RS/fold_3.py | 2 +-
.../tbx11k_simplified_v3_RS/fold_4.py | 2 +-
.../tbx11k_simplified_v3_RS/fold_5.py | 2 +-
.../tbx11k_simplified_v3_RS/fold_6.py | 2 +-
.../tbx11k_simplified_v3_RS/fold_7.py | 2 +-
.../tbx11k_simplified_v3_RS/fold_8.py | 2 +-
.../tbx11k_simplified_v3_RS/fold_9.py | 2 +-
src/ptbench/data/loader.py | 1 +
.../data/tbx11k_simplified/__init__.py | 7 +-
.../data/tbx11k_simplified_RS/__init__.py | 8 +-
.../data/tbx11k_simplified_v2/__init__.py | 7 +-
.../data/tbx11k_simplified_v2_RS/__init__.py | 8 +-
.../data/tbx11k_simplified_v3/__init__.py | 7 +-
.../data/tbx11k_simplified_v3_RS/__init__.py | 8 +-
tests/test_11k.py | 9 +-
tests/test_11k_v2.py | 9 +-
tests/test_11k_v3.py | 9 +-
tests/test_mc_ch_in_11k.py | 143 ++++++---
tests/test_mc_ch_in_11k_RS.py | 286 ++++++++++++------
tests/test_mc_ch_in_11kv2.py | 223 ++++++++++----
tests/test_mc_ch_in_11kv2_RS.py | 286 ++++++++++++------
140 files changed, 999 insertions(+), 536 deletions(-)
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/__init__.py b/src/ptbench/configs/datasets/mc_ch_in_11k/__init__.py
index df4ac6bb..e8970bd7 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/__init__.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/__init__.py
@@ -124,17 +124,34 @@ def _maker(protocol):
dataset = {}
dataset["__train__"] = ConcatDataset(
- [mc["__train__"], ch["__train__"], indian["__train__"], tbx11k["__train__"]]
+ [
+ mc["__train__"],
+ ch["__train__"],
+ indian["__train__"],
+ tbx11k["__train__"],
+ ]
)
dataset["train"] = ConcatDataset(
[mc["train"], ch["train"], indian["train"], tbx11k["train"]]
)
dataset["__valid__"] = ConcatDataset(
- [mc["__valid__"], ch["__valid__"], indian["__valid__"], tbx11k["__valid__"]]
+ [
+ mc["__valid__"],
+ ch["__valid__"],
+ indian["__valid__"],
+ tbx11k["__valid__"],
+ ]
)
dataset["validation"] = ConcatDataset(
- [mc["validation"], ch["validation"], indian["validation"], tbx11k["validation"]]
+ [
+ mc["validation"],
+ ch["validation"],
+ indian["validation"],
+ tbx11k["validation"],
+ ]
+ )
+ dataset["test"] = ConcatDataset(
+ [mc["test"], ch["test"], indian["test"], tbx11k["test"]]
)
- dataset["test"] = ConcatDataset([mc["test"], ch["test"], indian["test"], tbx11k["test"]])
return dataset
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/default.py b/src/ptbench/configs/datasets/mc_ch_in_11k/default.py
index 224d3aca..7d4f16bd 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/default.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/default.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets."""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets."""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0.py
index eeb11f2c..757a0eb9 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 0)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 0)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0_rgb.py
index d65df7b5..48e05ff3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 0, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 0, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1.py
index 6cb3cf81..56579589 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 1)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 1)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1_rgb.py
index 548fdbf2..c782d68d 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 1, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 1, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2.py
index 474c3f15..10a597bc 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 2)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 2)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2_rgb.py
index c303c093..d624f3af 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 2, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 2, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3.py
index b2b4d9ce..39bee4fe 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 3)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 3)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3_rgb.py
index 4c7f0865..7b26e425 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 3, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 3, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4.py
index bc39e7a3..5fb56292 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 4)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 4)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4_rgb.py
index cf86e941..fbc4f0cf 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 4, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 4, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5.py
index 1ca82222..679bb9b3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 5)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 5)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5_rgb.py
index d2821015..747d510e 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 5, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 5, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6.py
index d40c95f6..cd8e4cd5 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 6)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 6)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6_rgb.py
index 42c0b193..86f112c3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 6, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 6, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7.py
index 2b3d9b44..98241531 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 7)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 7)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7_rgb.py
index c32c4b25..981fe191 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 7, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 7, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8.py
index 2ccfbafe..dab1a234 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 8)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 8)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8_rgb.py
index c1dfad04..798b8de6 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 8, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 8, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9.py
index 07068cb8..097724b9 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 9)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 9)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9_rgb.py
index fba36c4b..c6c564a4 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 9, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 9, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/rgb.py
index 656a8c0f..f47796a8 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k/rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k/rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/__init__.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/__init__.py
index 3a0c4478..7fa4f307 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/__init__.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/__init__.py
@@ -124,17 +124,34 @@ def _maker(protocol):
dataset = {}
dataset["__train__"] = ConcatDataset(
- [mc["__train__"], ch["__train__"], indian["__train__"], tbx11k["__train__"]]
+ [
+ mc["__train__"],
+ ch["__train__"],
+ indian["__train__"],
+ tbx11k["__train__"],
+ ]
)
dataset["train"] = ConcatDataset(
[mc["train"], ch["train"], indian["train"], tbx11k["train"]]
)
dataset["__valid__"] = ConcatDataset(
- [mc["__valid__"], ch["__valid__"], indian["__valid__"], tbx11k["__valid__"]]
+ [
+ mc["__valid__"],
+ ch["__valid__"],
+ indian["__valid__"],
+ tbx11k["__valid__"],
+ ]
)
dataset["validation"] = ConcatDataset(
- [mc["validation"], ch["validation"], indian["validation"], tbx11k["validation"]]
+ [
+ mc["validation"],
+ ch["validation"],
+ indian["validation"],
+ tbx11k["validation"],
+ ]
+ )
+ dataset["test"] = ConcatDataset(
+ [mc["test"], ch["test"], indian["test"], tbx11k["test"]]
)
- dataset["test"] = ConcatDataset([mc["test"], ch["test"], indian["test"], tbx11k["test"]])
return dataset
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/default.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/default.py
index 224d3aca..7d4f16bd 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/default.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/default.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets."""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets."""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0.py
index eeb11f2c..757a0eb9 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 0)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 0)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0_rgb.py
index d65df7b5..48e05ff3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 0, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 0, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1.py
index 6cb3cf81..56579589 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 1)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 1)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1_rgb.py
index 548fdbf2..c782d68d 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 1, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 1, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2.py
index 474c3f15..10a597bc 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 2)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 2)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2_rgb.py
index c303c093..d624f3af 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 2, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 2, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3.py
index b2b4d9ce..39bee4fe 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 3)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 3)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3_rgb.py
index 4c7f0865..7b26e425 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 3, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 3, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4.py
index bc39e7a3..5fb56292 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 4)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 4)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4_rgb.py
index cf86e941..fbc4f0cf 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 4, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 4, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5.py
index 1ca82222..679bb9b3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 5)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 5)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5_rgb.py
index d2821015..747d510e 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 5, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 5, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6.py
index d40c95f6..cd8e4cd5 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 6)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 6)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6_rgb.py
index 42c0b193..86f112c3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 6, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 6, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7.py
index 2b3d9b44..98241531 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 7)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 7)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7_rgb.py
index c32c4b25..981fe191 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 7, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 7, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8.py
index 2ccfbafe..dab1a234 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 8)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 8)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8_rgb.py
index c1dfad04..798b8de6 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 8, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 8, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9.py
index 07068cb8..097724b9 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 9)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 9)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9_rgb.py
index fba36c4b..c6c564a4 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 9, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 9, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/rgb.py
index 656a8c0f..f47796a8 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/__init__.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/__init__.py
index 0789abe7..c36f7f60 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/__init__.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/__init__.py
@@ -124,17 +124,34 @@ def _maker(protocol):
dataset = {}
dataset["__train__"] = ConcatDataset(
- [mc["__train__"], ch["__train__"], indian["__train__"], tbx11kv2["__train__"]]
+ [
+ mc["__train__"],
+ ch["__train__"],
+ indian["__train__"],
+ tbx11kv2["__train__"],
+ ]
)
dataset["train"] = ConcatDataset(
[mc["train"], ch["train"], indian["train"], tbx11kv2["train"]]
)
dataset["__valid__"] = ConcatDataset(
- [mc["__valid__"], ch["__valid__"], indian["__valid__"], tbx11kv2["__valid__"]]
+ [
+ mc["__valid__"],
+ ch["__valid__"],
+ indian["__valid__"],
+ tbx11kv2["__valid__"],
+ ]
)
dataset["validation"] = ConcatDataset(
- [mc["validation"], ch["validation"], indian["validation"], tbx11kv2["validation"]]
+ [
+ mc["validation"],
+ ch["validation"],
+ indian["validation"],
+ tbx11kv2["validation"],
+ ]
+ )
+ dataset["test"] = ConcatDataset(
+ [mc["test"], ch["test"], indian["test"], tbx11kv2["test"]]
)
- dataset["test"] = ConcatDataset([mc["test"], ch["test"], indian["test"], tbx11kv2["test"]])
return dataset
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/default.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/default.py
index 224d3aca..7d4f16bd 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/default.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/default.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets."""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets."""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0.py
index eeb11f2c..757a0eb9 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 0)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 0)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0_rgb.py
index d65df7b5..48e05ff3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 0, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 0, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1.py
index 6cb3cf81..56579589 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 1)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 1)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1_rgb.py
index 548fdbf2..c782d68d 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 1, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 1, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2.py
index 474c3f15..10a597bc 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 2)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 2)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2_rgb.py
index c303c093..d624f3af 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 2, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 2, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3.py
index b2b4d9ce..39bee4fe 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 3)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 3)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3_rgb.py
index 4c7f0865..7b26e425 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 3, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 3, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4.py
index bc39e7a3..5fb56292 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 4)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 4)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4_rgb.py
index cf86e941..fbc4f0cf 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 4, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 4, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5.py
index 1ca82222..679bb9b3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 5)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 5)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5_rgb.py
index d2821015..747d510e 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 5, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 5, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6.py
index d40c95f6..cd8e4cd5 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 6)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 6)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6_rgb.py
index 42c0b193..86f112c3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 6, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 6, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7.py
index 2b3d9b44..98241531 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 7)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 7)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7_rgb.py
index c32c4b25..981fe191 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 7, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 7, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8.py
index 2ccfbafe..dab1a234 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 8)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 8)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8_rgb.py
index c1dfad04..798b8de6 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 8, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 8, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9.py
index 07068cb8..097724b9 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 9)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 9)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9_rgb.py
index fba36c4b..c6c564a4 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 9, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 9, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/rgb.py
index 656a8c0f..f47796a8 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2/rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/__init__.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/__init__.py
index 06725740..5e4b8276 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/__init__.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/__init__.py
@@ -124,17 +124,34 @@ def _maker(protocol):
dataset = {}
dataset["__train__"] = ConcatDataset(
- [mc["__train__"], ch["__train__"], indian["__train__"], tbx11kv2["__train__"]]
+ [
+ mc["__train__"],
+ ch["__train__"],
+ indian["__train__"],
+ tbx11kv2["__train__"],
+ ]
)
dataset["train"] = ConcatDataset(
[mc["train"], ch["train"], indian["train"], tbx11kv2["train"]]
)
dataset["__valid__"] = ConcatDataset(
- [mc["__valid__"], ch["__valid__"], indian["__valid__"], tbx11kv2["__valid__"]]
+ [
+ mc["__valid__"],
+ ch["__valid__"],
+ indian["__valid__"],
+ tbx11kv2["__valid__"],
+ ]
)
dataset["validation"] = ConcatDataset(
- [mc["validation"], ch["validation"], indian["validation"], tbx11kv2["validation"]]
+ [
+ mc["validation"],
+ ch["validation"],
+ indian["validation"],
+ tbx11kv2["validation"],
+ ]
+ )
+ dataset["test"] = ConcatDataset(
+ [mc["test"], ch["test"], indian["test"], tbx11kv2["test"]]
)
- dataset["test"] = ConcatDataset([mc["test"], ch["test"], indian["test"], tbx11kv2["test"]])
return dataset
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/default.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/default.py
index 224d3aca..7d4f16bd 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/default.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/default.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets."""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets."""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0.py
index eeb11f2c..757a0eb9 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 0)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 0)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0_rgb.py
index d65df7b5..48e05ff3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 0, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 0, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1.py
index 6cb3cf81..56579589 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 1)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 1)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1_rgb.py
index 548fdbf2..c782d68d 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 1, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 1, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2.py
index 474c3f15..10a597bc 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 2)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 2)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2_rgb.py
index c303c093..d624f3af 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 2, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 2, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3.py
index b2b4d9ce..39bee4fe 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 3)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 3)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3_rgb.py
index 4c7f0865..7b26e425 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 3, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 3, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4.py
index bc39e7a3..5fb56292 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 4)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 4)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4_rgb.py
index cf86e941..fbc4f0cf 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 4, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 4, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5.py
index 1ca82222..679bb9b3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 5)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 5)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5_rgb.py
index d2821015..747d510e 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 5, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 5, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6.py
index d40c95f6..cd8e4cd5 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 6)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 6)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6_rgb.py
index 42c0b193..86f112c3 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 6, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 6, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7.py
index 2b3d9b44..98241531 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 7)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 7)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7_rgb.py
index c32c4b25..981fe191 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 7, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 7, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8.py
index 2ccfbafe..dab1a234 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 8)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 8)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8_rgb.py
index c1dfad04..798b8de6 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 8, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 8, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9.py
index 07068cb8..097724b9 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 9)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 9)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9_rgb.py
index fba36c4b..c6c564a4 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9_rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9_rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (cross validation fold 9, RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (cross validation fold 9, RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/rgb.py
index 656a8c0f..f47796a8 100644
--- a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/rgb.py
+++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/rgb.py
@@ -2,8 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and
-the default TBX11K-simplified datasets (RGB)"""
+"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and the default
+TBX11K-simplified datasets (RGB)"""
from . import _maker
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v2/rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2/rgb.py
index 1af86d94..9614b847 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v2/rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v2/rgb.py
@@ -2,7 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""TBX11k simplified dataset for TB detection (default protocol, converted in RGB)
+"""TBX11k simplified dataset for TB detection (default protocol, converted in
+RGB)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/default.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/default.py
index 7208fc7d..3805a142 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/default.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/default.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_0.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_0.py
index 20c7316f..bed7342d 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_0.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_0.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_0_rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_0_rgb.py
index ce57f3bc..ed2f6d17 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_0_rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_0_rgb.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_1.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_1.py
index 4d3c1961..92cc9ce3 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_1.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_1.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_1_rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_1_rgb.py
index 227af634..9380ec5b 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_1_rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_1_rgb.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_2.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_2.py
index bc8088a0..460c96df 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_2.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_2.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_2_rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_2_rgb.py
index 1ee2b330..69641e3a 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_2_rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_2_rgb.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_3.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_3.py
index 3fba1cda..c799c7cb 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_3.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_3.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_3_rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_3_rgb.py
index 75ac8cd3..816eeca4 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_3_rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_3_rgb.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_4.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_4.py
index d5917fef..acceda91 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_4.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_4.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_4_rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_4_rgb.py
index 130539f9..07811e9c 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_4_rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_4_rgb.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_5.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_5.py
index 2cf02786..20b048d5 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_5.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_5.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_5_rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_5_rgb.py
index ab5dc438..6b851bdc 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_5_rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_5_rgb.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_6.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_6.py
index c11e5b2f..e99afb4c 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_6.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_6.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_6_rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_6_rgb.py
index 4785cfc3..01a94e08 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_6_rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_6_rgb.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_7.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_7.py
index d9cbb566..53e5af59 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_7.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_7.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_7_rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_7_rgb.py
index 6d914cee..546d6358 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_7_rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_7_rgb.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_8.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_8.py
index 5592912a..d59887df 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_8.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_8.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_8_rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_8_rgb.py
index 6db0e802..243c1455 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_8_rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_8_rgb.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_9.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_9.py
index 79410450..2f9a45f0 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_9.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_9.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_9_rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_9_rgb.py
index 82af3049..ef3d5045 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_9_rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/fold_9_rgb.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3/rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3/rgb.py
index 9729d510..1c274f79 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3/rgb.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3/rgb.py
@@ -2,11 +2,12 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""TBX11k simplified dataset for TB detection (default protocol, converted in RGB)
+"""TBX11k simplified dataset for TB detection (default protocol, converted in
+RGB)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/default.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/default.py
index 088f9186..7c62887d 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/default.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/default.py
@@ -7,7 +7,7 @@ DensenetRS predictions)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_0.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_0.py
index 20c7316f..bed7342d 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_0.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_0.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_1.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_1.py
index 4d3c1961..92cc9ce3 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_1.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_1.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_2.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_2.py
index bc8088a0..460c96df 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_2.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_2.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_3.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_3.py
index 3fba1cda..c799c7cb 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_3.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_3.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_4.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_4.py
index d5917fef..acceda91 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_4.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_4.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_5.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_5.py
index 2cf02786..20b048d5 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_5.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_5.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_6.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_6.py
index c11e5b2f..e99afb4c 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_6.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_6.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_7.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_7.py
index d9cbb566..53e5af59 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_7.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_7.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_8.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_8.py
index 5592912a..d59887df 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_8.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_8.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_9.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_9.py
index 79410450..2f9a45f0 100644
--- a/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_9.py
+++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_9.py
@@ -6,7 +6,7 @@
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
-* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
+* This split consists of the 4 labels "healthy", "latent TB", "sick & non-TB",
* and "active TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v3` for dataset details
diff --git a/src/ptbench/data/loader.py b/src/ptbench/data/loader.py
index 0ee253a7..12a7517e 100644
--- a/src/ptbench/data/loader.py
+++ b/src/ptbench/data/loader.py
@@ -102,6 +102,7 @@ def make_delayed(sample, loader, key=None):
label=sample["label"],
)
+
def make_delayed_bbox(sample, loader, key=None):
"""Returns a delayed-loading Sample object.
diff --git a/src/ptbench/data/tbx11k_simplified/__init__.py b/src/ptbench/data/tbx11k_simplified/__init__.py
index 46520581..dd2be35d 100644
--- a/src/ptbench/data/tbx11k_simplified/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified/__init__.py
@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""TBX11K simplified dataset for computer-aided diagnosis
+"""TBX11K simplified dataset for computer-aided diagnosis.
The TBX11K database has been established to foster research
in computer-aided diagnosis of pulmonary diseases with a special
@@ -48,12 +48,14 @@ _datadir = load_rc().get(
"datadir.tbx11k_simplified", os.path.realpath(os.curdir)
)
+
def _raw_data_loader(sample):
return dict(
data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
label=sample["label"],
)
+
def _raw_data_loader_bbox(sample):
return dict(
data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
@@ -61,16 +63,19 @@ def _raw_data_loader_bbox(sample):
bboxes=sample["bboxes"],
)
+
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader)
+
def _loader_bbox(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
return make_delayed_bbox(sample, _raw_data_loader_bbox)
+
dataset = JSONDataset(
protocols=_protocols,
fieldnames=("data", "label"),
diff --git a/src/ptbench/data/tbx11k_simplified_RS/__init__.py b/src/ptbench/data/tbx11k_simplified_RS/__init__.py
index 1f328703..5a7aa0d7 100644
--- a/src/ptbench/data/tbx11k_simplified_RS/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified_RS/__init__.py
@@ -47,6 +47,7 @@ _protocols = [
def _raw_data_loader(sample):
return dict(data=sample["data"], label=sample["label"])
+
def _raw_data_loader_bbox(sample):
return dict(
data=sample["data"],
@@ -54,15 +55,20 @@ def _raw_data_loader_bbox(sample):
bboxes=sample["bboxes"],
)
+
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we returned delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader, key=sample["filename"])
+
def _loader_bbox(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
- return make_delayed_bbox(sample, _raw_data_loader_bbox, key=sample["filename"])
+ return make_delayed_bbox(
+ sample, _raw_data_loader_bbox, key=sample["filename"]
+ )
+
dataset = JSONDataset(
protocols=_protocols,
diff --git a/src/ptbench/data/tbx11k_simplified_v2/__init__.py b/src/ptbench/data/tbx11k_simplified_v2/__init__.py
index 9cb5b281..7cf676b0 100644
--- a/src/ptbench/data/tbx11k_simplified_v2/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified_v2/__init__.py
@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""TBX11K simplified dataset for computer-aided diagnosis
+"""TBX11K simplified dataset for computer-aided diagnosis.
The TBX11K database has been established to foster research
in computer-aided diagnosis of pulmonary diseases with a special
@@ -48,12 +48,14 @@ _datadir = load_rc().get(
"datadir.tbx11k_simplified", os.path.realpath(os.curdir)
)
+
def _raw_data_loader(sample):
return dict(
data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
label=sample["label"],
)
+
def _raw_data_loader_bbox(sample):
return dict(
data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
@@ -61,16 +63,19 @@ def _raw_data_loader_bbox(sample):
bboxes=sample["bboxes"],
)
+
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader)
+
def _loader_bbox(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
return make_delayed_bbox(sample, _raw_data_loader_bbox)
+
dataset = JSONDataset(
protocols=_protocols,
fieldnames=("data", "label"),
diff --git a/src/ptbench/data/tbx11k_simplified_v2_RS/__init__.py b/src/ptbench/data/tbx11k_simplified_v2_RS/__init__.py
index 23dd76b7..31c0a4a5 100644
--- a/src/ptbench/data/tbx11k_simplified_v2_RS/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified_v2_RS/__init__.py
@@ -47,6 +47,7 @@ _protocols = [
def _raw_data_loader(sample):
return dict(data=sample["data"], label=sample["label"])
+
def _raw_data_loader_bbox(sample):
return dict(
data=sample["data"],
@@ -54,15 +55,20 @@ def _raw_data_loader_bbox(sample):
bboxes=sample["bboxes"],
)
+
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we returned delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader, key=sample["filename"])
+
def _loader_bbox(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
- return make_delayed_bbox(sample, _raw_data_loader_bbox, key=sample["filename"])
+ return make_delayed_bbox(
+ sample, _raw_data_loader_bbox, key=sample["filename"]
+ )
+
dataset = JSONDataset(
protocols=_protocols,
diff --git a/src/ptbench/data/tbx11k_simplified_v3/__init__.py b/src/ptbench/data/tbx11k_simplified_v3/__init__.py
index 9cb5b281..7cf676b0 100644
--- a/src/ptbench/data/tbx11k_simplified_v3/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified_v3/__init__.py
@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
-"""TBX11K simplified dataset for computer-aided diagnosis
+"""TBX11K simplified dataset for computer-aided diagnosis.
The TBX11K database has been established to foster research
in computer-aided diagnosis of pulmonary diseases with a special
@@ -48,12 +48,14 @@ _datadir = load_rc().get(
"datadir.tbx11k_simplified", os.path.realpath(os.curdir)
)
+
def _raw_data_loader(sample):
return dict(
data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
label=sample["label"],
)
+
def _raw_data_loader_bbox(sample):
return dict(
data=load_pil_baw(os.path.join(_datadir, sample["data"])), # type: ignore
@@ -61,16 +63,19 @@ def _raw_data_loader_bbox(sample):
bboxes=sample["bboxes"],
)
+
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader)
+
def _loader_bbox(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
return make_delayed_bbox(sample, _raw_data_loader_bbox)
+
dataset = JSONDataset(
protocols=_protocols,
fieldnames=("data", "label"),
diff --git a/src/ptbench/data/tbx11k_simplified_v3_RS/__init__.py b/src/ptbench/data/tbx11k_simplified_v3_RS/__init__.py
index 23dd76b7..31c0a4a5 100644
--- a/src/ptbench/data/tbx11k_simplified_v3_RS/__init__.py
+++ b/src/ptbench/data/tbx11k_simplified_v3_RS/__init__.py
@@ -47,6 +47,7 @@ _protocols = [
def _raw_data_loader(sample):
return dict(data=sample["data"], label=sample["label"])
+
def _raw_data_loader_bbox(sample):
return dict(
data=sample["data"],
@@ -54,15 +55,20 @@ def _raw_data_loader_bbox(sample):
bboxes=sample["bboxes"],
)
+
def _loader(context, sample):
# "context" is ignored in this case - database is homogeneous
# we returned delayed samples to avoid loading all images at once
return make_delayed(sample, _raw_data_loader, key=sample["filename"])
+
def _loader_bbox(context, sample):
# "context" is ignored in this case - database is homogeneous
# we return delayed samples to avoid loading all images at once
- return make_delayed_bbox(sample, _raw_data_loader_bbox, key=sample["filename"])
+ return make_delayed_bbox(
+ sample, _raw_data_loader_bbox, key=sample["filename"]
+ )
+
dataset = JSONDataset(
protocols=_protocols,
diff --git a/tests/test_11k.py b/tests/test_11k.py
index c8d7405c..ad7c0935 100644
--- a/tests/test_11k.py
+++ b/tests/test_11k.py
@@ -69,6 +69,7 @@ def test_protocol_consistency():
for s in subset["test"]:
assert s.label in [0.0, 1.0]
+
def test_protocol_consistency_bbox():
from ptbench.data.tbx11k_simplified import dataset_with_bboxes
@@ -139,6 +140,7 @@ def test_protocol_consistency_bbox():
for s in subset["train"]:
assert s.bboxes == "none" or s.bboxes[0].startswith("{'xmin':")
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified")
def test_loading():
from ptbench.data.tbx11k_simplified import dataset
@@ -162,6 +164,7 @@ def test_loading():
for s in subset["train"][:limit]:
_check_sample(s)
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified")
def test_loading_bbox():
from ptbench.data.tbx11k_simplified import dataset_with_bboxes
@@ -180,7 +183,9 @@ def test_loading_bbox():
assert data["label"] in [0, 1] # Check labels
assert "bboxes" in data
- assert data["bboxes"] == "none" or data["bboxes"][0].startswith("{'xmin':")
+ assert data["bboxes"] == "none" or data["bboxes"][0].startswith(
+ "{'xmin':"
+ )
limit = 30 # use this to limit testing to first images only, else None
@@ -188,12 +193,14 @@ def test_loading_bbox():
for s in subset["train"][:limit]:
_check_sample(s)
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified")
def test_check():
from ptbench.data.tbx11k_simplified import dataset
assert dataset.check() == 0
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified")
def test_check_bbox():
from ptbench.data.tbx11k_simplified import dataset_with_bboxes
diff --git a/tests/test_11k_v2.py b/tests/test_11k_v2.py
index cef3df72..48bc6823 100644
--- a/tests/test_11k_v2.py
+++ b/tests/test_11k_v2.py
@@ -69,6 +69,7 @@ def test_protocol_consistency():
for s in subset["test"]:
assert s.label in [0.0, 1.0]
+
def test_protocol_consistency_bbox():
from ptbench.data.tbx11k_simplified_v2 import dataset_with_bboxes
@@ -139,6 +140,7 @@ def test_protocol_consistency_bbox():
for s in subset["train"]:
assert s.bboxes == "none" or s.bboxes[0].startswith("{'xmin':")
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified_v2")
def test_loading():
from ptbench.data.tbx11k_simplified_v2 import dataset
@@ -162,6 +164,7 @@ def test_loading():
for s in subset["train"][:limit]:
_check_sample(s)
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified_v2")
def test_loading_bbox():
from ptbench.data.tbx11k_simplified_v2 import dataset_with_bboxes
@@ -180,7 +183,9 @@ def test_loading_bbox():
assert data["label"] in [0, 1] # Check labels
assert "bboxes" in data
- assert data["bboxes"] == "none" or data["bboxes"][0].startswith("{'xmin':")
+ assert data["bboxes"] == "none" or data["bboxes"][0].startswith(
+ "{'xmin':"
+ )
limit = 30 # use this to limit testing to first images only, else None
@@ -188,12 +193,14 @@ def test_loading_bbox():
for s in subset["train"][:limit]:
_check_sample(s)
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified_v2")
def test_check():
from ptbench.data.tbx11k_simplified_v2 import dataset
assert dataset.check() == 0
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified_v2")
def test_check_bbox():
from ptbench.data.tbx11k_simplified_v2 import dataset_with_bboxes
diff --git a/tests/test_11k_v3.py b/tests/test_11k_v3.py
index f659c408..a81af201 100644
--- a/tests/test_11k_v3.py
+++ b/tests/test_11k_v3.py
@@ -69,6 +69,7 @@ def test_protocol_consistency():
for s in subset["test"]:
assert s.label in [0.0, 1.0]
+
def test_protocol_consistency_bbox():
from ptbench.data.tbx11k_simplified_v3 import dataset_with_bboxes
@@ -139,6 +140,7 @@ def test_protocol_consistency_bbox():
for s in subset["train"]:
assert s.bboxes == "none" or s.bboxes[0].startswith("{'xmin':")
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified_v3")
def test_loading():
from ptbench.data.tbx11k_simplified_v3 import dataset
@@ -162,6 +164,7 @@ def test_loading():
for s in subset["train"][:limit]:
_check_sample(s)
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified_v3")
def test_loading_bbox():
from ptbench.data.tbx11k_simplified_v3 import dataset_with_bboxes
@@ -180,7 +183,9 @@ def test_loading_bbox():
assert data["label"] in [0, 1] # Check labels
assert "bboxes" in data
- assert data["bboxes"] == "none" or data["bboxes"][0].startswith("{'xmin':")
+ assert data["bboxes"] == "none" or data["bboxes"][0].startswith(
+ "{'xmin':"
+ )
limit = 30 # use this to limit testing to first images only, else None
@@ -188,12 +193,14 @@ def test_loading_bbox():
for s in subset["train"][:limit]:
_check_sample(s)
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified_v3")
def test_check():
from ptbench.data.tbx11k_simplified_v3 import dataset
assert dataset.check() == 0
+
@pytest.mark.skip_if_rc_var_not_set("datadir.tbx11k_simplified_v3")
def test_check_bbox():
from ptbench.data.tbx11k_simplified_v3 import dataset_with_bboxes
diff --git a/tests/test_mc_ch_in_11k.py b/tests/test_mc_ch_in_11k.py
index a2ede3d9..9aeb0c36 100644
--- a/tests/test_mc_ch_in_11k.py
+++ b/tests/test_mc_ch_in_11k.py
@@ -29,25 +29,45 @@ def test_dataset_consistency():
from ptbench.configs.datasets.indian import fold_9_rgb as indian_f9_rgb
from ptbench.configs.datasets.mc_ch_in_11k import default as mc_ch_in_11k
from ptbench.configs.datasets.mc_ch_in_11k import fold_0 as mc_ch_in_11k_f0
- from ptbench.configs.datasets.mc_ch_in_11k import fold_0_rgb as mc_ch_in_11k_f0_rgb
+ from ptbench.configs.datasets.mc_ch_in_11k import (
+ fold_0_rgb as mc_ch_in_11k_f0_rgb,
+ )
from ptbench.configs.datasets.mc_ch_in_11k import fold_1 as mc_ch_in_11k_f1
- from ptbench.configs.datasets.mc_ch_in_11k import fold_1_rgb as mc_ch_in_11k_f1_rgb
+ from ptbench.configs.datasets.mc_ch_in_11k import (
+ fold_1_rgb as mc_ch_in_11k_f1_rgb,
+ )
from ptbench.configs.datasets.mc_ch_in_11k import fold_2 as mc_ch_in_11k_f2
- from ptbench.configs.datasets.mc_ch_in_11k import fold_2_rgb as mc_ch_in_11k_f2_rgb
+ from ptbench.configs.datasets.mc_ch_in_11k import (
+ fold_2_rgb as mc_ch_in_11k_f2_rgb,
+ )
from ptbench.configs.datasets.mc_ch_in_11k import fold_3 as mc_ch_in_11k_f3
- from ptbench.configs.datasets.mc_ch_in_11k import fold_3_rgb as mc_ch_in_11k_f3_rgb
+ from ptbench.configs.datasets.mc_ch_in_11k import (
+ fold_3_rgb as mc_ch_in_11k_f3_rgb,
+ )
from ptbench.configs.datasets.mc_ch_in_11k import fold_4 as mc_ch_in_11k_f4
- from ptbench.configs.datasets.mc_ch_in_11k import fold_4_rgb as mc_ch_in_11k_f4_rgb
+ from ptbench.configs.datasets.mc_ch_in_11k import (
+ fold_4_rgb as mc_ch_in_11k_f4_rgb,
+ )
from ptbench.configs.datasets.mc_ch_in_11k import fold_5 as mc_ch_in_11k_f5
- from ptbench.configs.datasets.mc_ch_in_11k import fold_5_rgb as mc_ch_in_11k_f5_rgb
+ from ptbench.configs.datasets.mc_ch_in_11k import (
+ fold_5_rgb as mc_ch_in_11k_f5_rgb,
+ )
from ptbench.configs.datasets.mc_ch_in_11k import fold_6 as mc_ch_in_11k_f6
- from ptbench.configs.datasets.mc_ch_in_11k import fold_6_rgb as mc_ch_in_11k_f6_rgb
+ from ptbench.configs.datasets.mc_ch_in_11k import (
+ fold_6_rgb as mc_ch_in_11k_f6_rgb,
+ )
from ptbench.configs.datasets.mc_ch_in_11k import fold_7 as mc_ch_in_11k_f7
- from ptbench.configs.datasets.mc_ch_in_11k import fold_7_rgb as mc_ch_in_11k_f7_rgb
+ from ptbench.configs.datasets.mc_ch_in_11k import (
+ fold_7_rgb as mc_ch_in_11k_f7_rgb,
+ )
from ptbench.configs.datasets.mc_ch_in_11k import fold_8 as mc_ch_in_11k_f8
- from ptbench.configs.datasets.mc_ch_in_11k import fold_8_rgb as mc_ch_in_11k_f8_rgb
+ from ptbench.configs.datasets.mc_ch_in_11k import (
+ fold_8_rgb as mc_ch_in_11k_f8_rgb,
+ )
from ptbench.configs.datasets.mc_ch_in_11k import fold_9 as mc_ch_in_11k_f9
- from ptbench.configs.datasets.mc_ch_in_11k import fold_9_rgb as mc_ch_in_11k_f9_rgb
+ from ptbench.configs.datasets.mc_ch_in_11k import (
+ fold_9_rgb as mc_ch_in_11k_f9_rgb,
+ )
from ptbench.configs.datasets.montgomery import default as mc
from ptbench.configs.datasets.montgomery import fold_0 as mc_f0
from ptbench.configs.datasets.montgomery import fold_0_rgb as mc_f0_rgb
@@ -92,25 +112,45 @@ def test_dataset_consistency():
from ptbench.configs.datasets.shenzhen import fold_9_rgb as ch_f9_rgb
from ptbench.configs.datasets.tbx11k_simplified import default as tbx11k
from ptbench.configs.datasets.tbx11k_simplified import fold_0 as tbx11k_f0
- from ptbench.configs.datasets.tbx11k_simplified import fold_0_rgb as tbx11k_f0_rgb
+ from ptbench.configs.datasets.tbx11k_simplified import (
+ fold_0_rgb as tbx11k_f0_rgb,
+ )
from ptbench.configs.datasets.tbx11k_simplified import fold_1 as tbx11k_f1
- from ptbench.configs.datasets.tbx11k_simplified import fold_1_rgb as tbx11k_f1_rgb
+ from ptbench.configs.datasets.tbx11k_simplified import (
+ fold_1_rgb as tbx11k_f1_rgb,
+ )
from ptbench.configs.datasets.tbx11k_simplified import fold_2 as tbx11k_f2
- from ptbench.configs.datasets.tbx11k_simplified import fold_2_rgb as tbx11k_f2_rgb
+ from ptbench.configs.datasets.tbx11k_simplified import (
+ fold_2_rgb as tbx11k_f2_rgb,
+ )
from ptbench.configs.datasets.tbx11k_simplified import fold_3 as tbx11k_f3
- from ptbench.configs.datasets.tbx11k_simplified import fold_3_rgb as tbx11k_f3_rgb
+ from ptbench.configs.datasets.tbx11k_simplified import (
+ fold_3_rgb as tbx11k_f3_rgb,
+ )
from ptbench.configs.datasets.tbx11k_simplified import fold_4 as tbx11k_f4
- from ptbench.configs.datasets.tbx11k_simplified import fold_4_rgb as tbx11k_f4_rgb
+ from ptbench.configs.datasets.tbx11k_simplified import (
+ fold_4_rgb as tbx11k_f4_rgb,
+ )
from ptbench.configs.datasets.tbx11k_simplified import fold_5 as tbx11k_f5
- from ptbench.configs.datasets.tbx11k_simplified import fold_5_rgb as tbx11k_f5_rgb
+ from ptbench.configs.datasets.tbx11k_simplified import (
+ fold_5_rgb as tbx11k_f5_rgb,
+ )
from ptbench.configs.datasets.tbx11k_simplified import fold_6 as tbx11k_f6
- from ptbench.configs.datasets.tbx11k_simplified import fold_6_rgb as tbx11k_f6_rgb
+ from ptbench.configs.datasets.tbx11k_simplified import (
+ fold_6_rgb as tbx11k_f6_rgb,
+ )
from ptbench.configs.datasets.tbx11k_simplified import fold_7 as tbx11k_f7
- from ptbench.configs.datasets.tbx11k_simplified import fold_7_rgb as tbx11k_f7_rgb
+ from ptbench.configs.datasets.tbx11k_simplified import (
+ fold_7_rgb as tbx11k_f7_rgb,
+ )
from ptbench.configs.datasets.tbx11k_simplified import fold_8 as tbx11k_f8
- from ptbench.configs.datasets.tbx11k_simplified import fold_8_rgb as tbx11k_f8_rgb
+ from ptbench.configs.datasets.tbx11k_simplified import (
+ fold_8_rgb as tbx11k_f8_rgb,
+ )
from ptbench.configs.datasets.tbx11k_simplified import fold_9 as tbx11k_f9
- from ptbench.configs.datasets.tbx11k_simplified import fold_9_rgb as tbx11k_f9_rgb
+ from ptbench.configs.datasets.tbx11k_simplified import (
+ fold_9_rgb as tbx11k_f9_rgb,
+ )
# Default protocol
mc_ch_in_11k_dataset = mc_ch_in_11k.dataset
@@ -130,7 +170,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -155,7 +196,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -180,7 +222,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -205,7 +248,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -230,7 +274,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -255,7 +300,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -280,7 +326,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -305,7 +352,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -330,7 +378,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -355,7 +404,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -380,7 +430,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -405,7 +456,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -430,7 +482,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -455,7 +508,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -480,7 +534,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -505,7 +560,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -530,7 +586,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -555,7 +612,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -580,7 +638,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -605,7 +664,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -630,7 +690,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
diff --git a/tests/test_mc_ch_in_11k_RS.py b/tests/test_mc_ch_in_11k_RS.py
index a453a55d..35fc4d88 100644
--- a/tests/test_mc_ch_in_11k_RS.py
+++ b/tests/test_mc_ch_in_11k_RS.py
@@ -17,17 +17,39 @@ def test_dataset_consistency():
from ptbench.configs.datasets.indian_RS import fold_7 as indian_f7
from ptbench.configs.datasets.indian_RS import fold_8 as indian_f8
from ptbench.configs.datasets.indian_RS import fold_9 as indian_f9
- from ptbench.configs.datasets.mc_ch_in_11k_RS import default as mc_ch_in_11k_RS
- from ptbench.configs.datasets.mc_ch_in_11k_RS import fold_0 as mc_ch_in_11k_f0
- from ptbench.configs.datasets.mc_ch_in_11k_RS import fold_1 as mc_ch_in_11k_f1
- from ptbench.configs.datasets.mc_ch_in_11k_RS import fold_2 as mc_ch_in_11k_f2
- from ptbench.configs.datasets.mc_ch_in_11k_RS import fold_3 as mc_ch_in_11k_f3
- from ptbench.configs.datasets.mc_ch_in_11k_RS import fold_4 as mc_ch_in_11k_f4
- from ptbench.configs.datasets.mc_ch_in_11k_RS import fold_5 as mc_ch_in_11k_f5
- from ptbench.configs.datasets.mc_ch_in_11k_RS import fold_6 as mc_ch_in_11k_f6
- from ptbench.configs.datasets.mc_ch_in_11k_RS import fold_7 as mc_ch_in_11k_f7
- from ptbench.configs.datasets.mc_ch_in_11k_RS import fold_8 as mc_ch_in_11k_f8
- from ptbench.configs.datasets.mc_ch_in_11k_RS import fold_9 as mc_ch_in_11k_f9
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ default as mc_ch_in_11k_RS,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ fold_0 as mc_ch_in_11k_f0,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ fold_1 as mc_ch_in_11k_f1,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ fold_2 as mc_ch_in_11k_f2,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ fold_3 as mc_ch_in_11k_f3,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ fold_4 as mc_ch_in_11k_f4,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ fold_5 as mc_ch_in_11k_f5,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ fold_6 as mc_ch_in_11k_f6,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ fold_7 as mc_ch_in_11k_f7,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ fold_8 as mc_ch_in_11k_f8,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11k_RS import (
+ fold_9 as mc_ch_in_11k_f9,
+ )
from ptbench.configs.datasets.montgomery_RS import default as mc_RS
from ptbench.configs.datasets.montgomery_RS import fold_0 as mc_f0
from ptbench.configs.datasets.montgomery_RS import fold_1 as mc_f1
@@ -50,17 +72,39 @@ def test_dataset_consistency():
from ptbench.configs.datasets.shenzhen_RS import fold_7 as ch_f7
from ptbench.configs.datasets.shenzhen_RS import fold_8 as ch_f8
from ptbench.configs.datasets.shenzhen_RS import fold_9 as ch_f9
- from ptbench.configs.datasets.tbx11k_simplified_RS import default as tbx11k_RS
- from ptbench.configs.datasets.tbx11k_simplified_RS import fold_0 as tbx11k_f0
- from ptbench.configs.datasets.tbx11k_simplified_RS import fold_1 as tbx11k_f1
- from ptbench.configs.datasets.tbx11k_simplified_RS import fold_2 as tbx11k_f2
- from ptbench.configs.datasets.tbx11k_simplified_RS import fold_3 as tbx11k_f3
- from ptbench.configs.datasets.tbx11k_simplified_RS import fold_4 as tbx11k_f4
- from ptbench.configs.datasets.tbx11k_simplified_RS import fold_5 as tbx11k_f5
- from ptbench.configs.datasets.tbx11k_simplified_RS import fold_6 as tbx11k_f6
- from ptbench.configs.datasets.tbx11k_simplified_RS import fold_7 as tbx11k_f7
- from ptbench.configs.datasets.tbx11k_simplified_RS import fold_8 as tbx11k_f8
- from ptbench.configs.datasets.tbx11k_simplified_RS import fold_9 as tbx11k_f9
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ default as tbx11k_RS,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ fold_0 as tbx11k_f0,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ fold_1 as tbx11k_f1,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ fold_2 as tbx11k_f2,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ fold_3 as tbx11k_f3,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ fold_4 as tbx11k_f4,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ fold_5 as tbx11k_f5,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ fold_6 as tbx11k_f6,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ fold_7 as tbx11k_f7,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ fold_8 as tbx11k_f8,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_RS import (
+ fold_9 as tbx11k_f9,
+ )
# Default protocol
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_RS.dataset
@@ -74,19 +118,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_RS_dataset["train"]
- ) + len(ch_RS_dataset["train"]) + len(in_RS_dataset["train"] + len(
- tbx11k_RS_dataset["train"]))
+ ) + len(ch_RS_dataset["train"]) + len(
+ in_RS_dataset["train"] + len(tbx11k_RS_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_RS_dataset["validation"]
- ) + len(ch_RS_dataset["validation"]) + len(in_RS_dataset["validation"] + len(
- tbx11k_RS_dataset["validation"]))
+ ) + len(ch_RS_dataset["validation"]) + len(
+ in_RS_dataset["validation"] + len(tbx11k_RS_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_RS_dataset["test"]) + len(
- ch_RS_dataset["test"]
- ) + len(in_RS_dataset["test"] + len(tbx11k_RS_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_RS_dataset["test"]
+ ) + len(ch_RS_dataset["test"]) + len(
+ in_RS_dataset["test"] + len(tbx11k_RS_dataset["test"])
+ )
# Fold 0
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f0.dataset
@@ -100,19 +148,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 1
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f1.dataset
@@ -126,19 +178,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 2
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f2.dataset
@@ -152,19 +208,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 3
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f3.dataset
@@ -178,19 +238,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 4
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f4.dataset
@@ -204,19 +268,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 5
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f5.dataset
@@ -230,19 +298,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 6
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f6.dataset
@@ -256,19 +328,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 7
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f7.dataset
@@ -282,19 +358,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 8
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f8.dataset
@@ -308,19 +388,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 9
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f9.dataset
@@ -334,16 +418,20 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
diff --git a/tests/test_mc_ch_in_11kv2.py b/tests/test_mc_ch_in_11kv2.py
index 8b09f8f8..c923a9f5 100644
--- a/tests/test_mc_ch_in_11kv2.py
+++ b/tests/test_mc_ch_in_11kv2.py
@@ -28,26 +28,66 @@ def test_dataset_consistency():
from ptbench.configs.datasets.indian import fold_9 as indian_f9
from ptbench.configs.datasets.indian import fold_9_rgb as indian_f9_rgb
from ptbench.configs.datasets.mc_ch_in_11kv2 import default as mc_ch_in_11k
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_0 as mc_ch_in_11k_f0
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_0_rgb as mc_ch_in_11k_f0_rgb
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_1 as mc_ch_in_11k_f1
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_1_rgb as mc_ch_in_11k_f1_rgb
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_2 as mc_ch_in_11k_f2
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_2_rgb as mc_ch_in_11k_f2_rgb
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_3 as mc_ch_in_11k_f3
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_3_rgb as mc_ch_in_11k_f3_rgb
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_4 as mc_ch_in_11k_f4
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_4_rgb as mc_ch_in_11k_f4_rgb
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_5 as mc_ch_in_11k_f5
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_5_rgb as mc_ch_in_11k_f5_rgb
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_6 as mc_ch_in_11k_f6
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_6_rgb as mc_ch_in_11k_f6_rgb
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_7 as mc_ch_in_11k_f7
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_7_rgb as mc_ch_in_11k_f7_rgb
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_8 as mc_ch_in_11k_f8
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_8_rgb as mc_ch_in_11k_f8_rgb
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_9 as mc_ch_in_11k_f9
- from ptbench.configs.datasets.mc_ch_in_11kv2 import fold_9_rgb as mc_ch_in_11k_f9_rgb
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_0 as mc_ch_in_11k_f0,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_0_rgb as mc_ch_in_11k_f0_rgb,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_1 as mc_ch_in_11k_f1,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_1_rgb as mc_ch_in_11k_f1_rgb,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_2 as mc_ch_in_11k_f2,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_2_rgb as mc_ch_in_11k_f2_rgb,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_3 as mc_ch_in_11k_f3,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_3_rgb as mc_ch_in_11k_f3_rgb,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_4 as mc_ch_in_11k_f4,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_4_rgb as mc_ch_in_11k_f4_rgb,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_5 as mc_ch_in_11k_f5,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_5_rgb as mc_ch_in_11k_f5_rgb,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_6 as mc_ch_in_11k_f6,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_6_rgb as mc_ch_in_11k_f6_rgb,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_7 as mc_ch_in_11k_f7,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_7_rgb as mc_ch_in_11k_f7_rgb,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_8 as mc_ch_in_11k_f8,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_8_rgb as mc_ch_in_11k_f8_rgb,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_9 as mc_ch_in_11k_f9,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2 import (
+ fold_9_rgb as mc_ch_in_11k_f9_rgb,
+ )
from ptbench.configs.datasets.montgomery import default as mc
from ptbench.configs.datasets.montgomery import fold_0 as mc_f0
from ptbench.configs.datasets.montgomery import fold_0_rgb as mc_f0_rgb
@@ -91,26 +131,66 @@ def test_dataset_consistency():
from ptbench.configs.datasets.shenzhen import fold_9 as ch_f9
from ptbench.configs.datasets.shenzhen import fold_9_rgb as ch_f9_rgb
from ptbench.configs.datasets.tbx11k_simplified_v2 import default as tbx11k
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_0 as tbx11k_f0
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_0_rgb as tbx11k_f0_rgb
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_1 as tbx11k_f1
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_1_rgb as tbx11k_f1_rgb
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_2 as tbx11k_f2
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_2_rgb as tbx11k_f2_rgb
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_3 as tbx11k_f3
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_3_rgb as tbx11k_f3_rgb
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_4 as tbx11k_f4
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_4_rgb as tbx11k_f4_rgb
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_5 as tbx11k_f5
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_5_rgb as tbx11k_f5_rgb
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_6 as tbx11k_f6
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_6_rgb as tbx11k_f6_rgb
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_7 as tbx11k_f7
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_7_rgb as tbx11k_f7_rgb
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_8 as tbx11k_f8
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_8_rgb as tbx11k_f8_rgb
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_9 as tbx11k_f9
- from ptbench.configs.datasets.tbx11k_simplified_v2 import fold_9_rgb as tbx11k_f9_rgb
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_0 as tbx11k_f0,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_0_rgb as tbx11k_f0_rgb,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_1 as tbx11k_f1,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_1_rgb as tbx11k_f1_rgb,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_2 as tbx11k_f2,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_2_rgb as tbx11k_f2_rgb,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_3 as tbx11k_f3,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_3_rgb as tbx11k_f3_rgb,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_4 as tbx11k_f4,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_4_rgb as tbx11k_f4_rgb,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_5 as tbx11k_f5,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_5_rgb as tbx11k_f5_rgb,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_6 as tbx11k_f6,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_6_rgb as tbx11k_f6_rgb,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_7 as tbx11k_f7,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_7_rgb as tbx11k_f7_rgb,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_8 as tbx11k_f8,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_8_rgb as tbx11k_f8_rgb,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_9 as tbx11k_f9,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2 import (
+ fold_9_rgb as tbx11k_f9_rgb,
+ )
# Default protocol
mc_ch_in_11k_dataset = mc_ch_in_11k.dataset
@@ -130,7 +210,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -155,7 +236,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -180,7 +262,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -205,7 +288,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -230,7 +314,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -255,7 +340,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -280,7 +366,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -305,7 +392,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -330,7 +418,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -355,7 +444,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -380,7 +470,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -405,7 +496,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -430,7 +522,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -455,7 +548,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -480,7 +574,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -505,7 +600,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -530,7 +626,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -555,7 +652,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -580,7 +678,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -605,7 +704,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
@@ -630,7 +730,8 @@ def test_dataset_consistency():
assert len(mc_ch_in_11k_dataset["validation"]) == len(
mc_dataset["validation"]
) + len(ch_dataset["validation"]) + len(in_dataset["validation"]) + len(
- tbx11k_dataset["validation"])
+ tbx11k_dataset["validation"]
+ )
assert "test" in mc_ch_in_11k_dataset
assert len(mc_ch_in_11k_dataset["test"]) == len(mc_dataset["test"]) + len(
diff --git a/tests/test_mc_ch_in_11kv2_RS.py b/tests/test_mc_ch_in_11kv2_RS.py
index f2223948..3ee663f2 100644
--- a/tests/test_mc_ch_in_11kv2_RS.py
+++ b/tests/test_mc_ch_in_11kv2_RS.py
@@ -17,17 +17,39 @@ def test_dataset_consistency():
from ptbench.configs.datasets.indian_RS import fold_7 as indian_f7
from ptbench.configs.datasets.indian_RS import fold_8 as indian_f8
from ptbench.configs.datasets.indian_RS import fold_9 as indian_f9
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import default as mc_ch_in_11k_RS
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import fold_0 as mc_ch_in_11k_f0
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import fold_1 as mc_ch_in_11k_f1
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import fold_2 as mc_ch_in_11k_f2
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import fold_3 as mc_ch_in_11k_f3
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import fold_4 as mc_ch_in_11k_f4
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import fold_5 as mc_ch_in_11k_f5
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import fold_6 as mc_ch_in_11k_f6
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import fold_7 as mc_ch_in_11k_f7
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import fold_8 as mc_ch_in_11k_f8
- from ptbench.configs.datasets.mc_ch_in_11kv2_RS import fold_9 as mc_ch_in_11k_f9
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ default as mc_ch_in_11k_RS,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ fold_0 as mc_ch_in_11k_f0,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ fold_1 as mc_ch_in_11k_f1,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ fold_2 as mc_ch_in_11k_f2,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ fold_3 as mc_ch_in_11k_f3,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ fold_4 as mc_ch_in_11k_f4,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ fold_5 as mc_ch_in_11k_f5,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ fold_6 as mc_ch_in_11k_f6,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ fold_7 as mc_ch_in_11k_f7,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ fold_8 as mc_ch_in_11k_f8,
+ )
+ from ptbench.configs.datasets.mc_ch_in_11kv2_RS import (
+ fold_9 as mc_ch_in_11k_f9,
+ )
from ptbench.configs.datasets.montgomery_RS import default as mc_RS
from ptbench.configs.datasets.montgomery_RS import fold_0 as mc_f0
from ptbench.configs.datasets.montgomery_RS import fold_1 as mc_f1
@@ -50,17 +72,39 @@ def test_dataset_consistency():
from ptbench.configs.datasets.shenzhen_RS import fold_7 as ch_f7
from ptbench.configs.datasets.shenzhen_RS import fold_8 as ch_f8
from ptbench.configs.datasets.shenzhen_RS import fold_9 as ch_f9
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import default as tbx11k_RS
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import fold_0 as tbx11k_f0
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import fold_1 as tbx11k_f1
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import fold_2 as tbx11k_f2
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import fold_3 as tbx11k_f3
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import fold_4 as tbx11k_f4
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import fold_5 as tbx11k_f5
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import fold_6 as tbx11k_f6
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import fold_7 as tbx11k_f7
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import fold_8 as tbx11k_f8
- from ptbench.configs.datasets.tbx11k_simplified_v2_RS import fold_9 as tbx11k_f9
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ default as tbx11k_RS,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ fold_0 as tbx11k_f0,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ fold_1 as tbx11k_f1,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ fold_2 as tbx11k_f2,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ fold_3 as tbx11k_f3,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ fold_4 as tbx11k_f4,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ fold_5 as tbx11k_f5,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ fold_6 as tbx11k_f6,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ fold_7 as tbx11k_f7,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ fold_8 as tbx11k_f8,
+ )
+ from ptbench.configs.datasets.tbx11k_simplified_v2_RS import (
+ fold_9 as tbx11k_f9,
+ )
# Default protocol
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_RS.dataset
@@ -74,19 +118,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_RS_dataset["train"]
- ) + len(ch_RS_dataset["train"]) + len(in_RS_dataset["train"] + len(
- tbx11k_RS_dataset["train"]))
+ ) + len(ch_RS_dataset["train"]) + len(
+ in_RS_dataset["train"] + len(tbx11k_RS_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_RS_dataset["validation"]
- ) + len(ch_RS_dataset["validation"]) + len(in_RS_dataset["validation"] + len(
- tbx11k_RS_dataset["validation"]))
+ ) + len(ch_RS_dataset["validation"]) + len(
+ in_RS_dataset["validation"] + len(tbx11k_RS_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_RS_dataset["test"]) + len(
- ch_RS_dataset["test"]
- ) + len(in_RS_dataset["test"] + len(tbx11k_RS_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_RS_dataset["test"]
+ ) + len(ch_RS_dataset["test"]) + len(
+ in_RS_dataset["test"] + len(tbx11k_RS_dataset["test"])
+ )
# Fold 0
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f0.dataset
@@ -100,19 +148,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 1
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f1.dataset
@@ -126,19 +178,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 2
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f2.dataset
@@ -152,19 +208,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 3
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f3.dataset
@@ -178,19 +238,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 4
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f4.dataset
@@ -204,19 +268,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 5
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f5.dataset
@@ -230,19 +298,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 6
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f6.dataset
@@ -256,19 +328,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 7
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f7.dataset
@@ -282,19 +358,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 8
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f8.dataset
@@ -308,19 +388,23 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
# Fold 9
mc_ch_in_11k_RS_dataset = mc_ch_in_11k_f9.dataset
@@ -334,16 +418,20 @@ def test_dataset_consistency():
assert "train" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["train"]) == len(
mc_dataset["train"]
- ) + len(ch_dataset["train"]) + len(in_dataset["train"] + len(
- tbx11k_dataset["train"]))
+ ) + len(ch_dataset["train"]) + len(
+ in_dataset["train"] + len(tbx11k_dataset["train"])
+ )
assert "validation" in mc_ch_in_11k_RS_dataset
assert len(mc_ch_in_11k_RS_dataset["validation"]) == len(
mc_dataset["validation"]
- ) + len(ch_dataset["validation"]) + len(in_dataset["validation"] + len(
- tbx11k_dataset["validation"]))
+ ) + len(ch_dataset["validation"]) + len(
+ in_dataset["validation"] + len(tbx11k_dataset["validation"])
+ )
assert "test" in mc_ch_in_11k_RS_dataset
- assert len(mc_ch_in_11k_RS_dataset["test"]) == len(mc_dataset["test"]) + len(
- ch_dataset["test"]
- ) + len(in_dataset["test"] + len(tbx11k_dataset["test"]))
+ assert len(mc_ch_in_11k_RS_dataset["test"]) == len(
+ mc_dataset["test"]
+ ) + len(ch_dataset["test"]) + len(
+ in_dataset["test"] + len(tbx11k_dataset["test"])
+ )
--
GitLab