Reviewed DataModule design+docs+types

In this MR, I try to review the use of the data module and go further in simplifications for the whole code:

• Slightly streamlined the datamodule approach
• Added documentation
• Added type annotations
• Added some TODOs for further discussion

Closes #7 (closed).

TODO:

• (@dcarron) Implement/document/type hint normaliser (ensure it is used correctly through all the models; integrate on the train script) and that train CLI works!
• (@andre.anjos) Deal with import reweight_BCEWithLogitsLoss

pyproject.toml

@@ -15,33 +15,33 @@ dynamic = ["readme"]
 license = { text = "GNU General Public License v3 (GPLv3)" }
 authors = [{ name = "Geoffrey Raposo", email = "geoffrey@raposo.ch" }]
 maintainers = [
-    { name = "Andre Anjos", email = "andre.anjos@idiap.ch" },
-    { name = "Daniel Carron", email = "daniel.carron@idiap.ch" },
+  { name = "Andre Anjos", email = "andre.anjos@idiap.ch" },
+  { name = "Daniel Carron", email = "daniel.carron@idiap.ch" },
 ]
 classifiers = [
-    "Development Status :: 4 - Beta",
-    "Intended Audience :: Developers",
-    "License :: OSI Approved :: GNU General Public License v3 (GPLv3)",
-    "Natural Language :: English",
-    "Programming Language :: Python :: 3",
-    "Topic :: Software Development :: Libraries :: Python Modules",
+  "Development Status :: 4 - Beta",
+  "Intended Audience :: Developers",
+  "License :: OSI Approved :: GNU General Public License v3 (GPLv3)",
+  "Natural Language :: English",
+  "Programming Language :: Python :: 3",
+  "Topic :: Software Development :: Libraries :: Python Modules",
 ]
 dependencies = [
-    "clapper",
-    "click",
-    "numpy",
-    "pandas",
-    "scipy",
-    "scikit-learn",
-    "tqdm",
-    "psutil",
-    "tabulate",
-    "matplotlib",
-    "pillow",
-    "torch>=1.8",
-    "torchvision>=0.10",
-    "lightning>=2.0.3",
-    "tensorboard",
+  "clapper",
+  "click",
+  "numpy",
+  "pandas",
+  "scipy",
+  "scikit-learn",
+  "tqdm",
+  "psutil",
+  "tabulate",
+  "matplotlib",
+  "pillow",
+  "torch>=1.8",
+  "torchvision>=0.10",
+  "lightning>=2.0.3",
+  "tensorboard",
 ]
 
 [project.urls]
@@ -53,13 +53,13 @@ changelog = "https://gitlab.idiap.ch/biosignal/software/ptbench/-/releases"
 [project.optional-dependencies]
 qa = ["pre-commit"]
 doc = [
-    "sphinx",
-    "furo",
-    "sphinx-autodoc-typehints",
-    "auto-intersphinx",
-    "sphinx-copybutton",
-    "sphinx-inline-tabs",
-    "sphinx-click",
+  "sphinx",
+  "furo",
+  "sphinx-autodoc-typehints",
+  "auto-intersphinx",
+  "sphinx-copybutton",
+  "sphinx-inline-tabs",
+  "sphinx-click",
 ]
 test = ["pytest", "pytest-cov", "coverage"]