docs changes

config/model_config.yaml

@@ -14,7 +14,7 @@ conditioning_vars: # for each desired conditioning variable, add the name and nu
   state: 3
 
 diffcharge:
-  batch_size: 4
+  batch_size: 64
   n_epochs: 1000
   init_lr: 3e-5
   network: cnn # attention

setup.py

@@ -27,10 +27,8 @@
     "tensorboardX>=2.6.2.2",
     "pyyaml>=6.0.1",
     "pre-commit>=3.5.0",
-    "mypy>=1.10.1",
     "black>=24.4.2",
     "isort>=5.13.2",
-    "autoflake>=2.3.1",
     "dtaidistance>=2.3.12",
     "seaborn>=0.13.2",
     "einops>=0.8.0",
@@ -53,10 +51,10 @@
     "pip>=9.0.1",
     "watchdog>=0.8.3",
     # docs
-    "m2r2>=0.2.0",
-    "Sphinx>=4.0.2,<6.0.0",
-    "sphinx_rtd_theme>=0.2.4,<0.5",
-    "autodocsumm>=0.1.10",
+    # "m2r2>=0.2.0",
+    # "Sphinx>=4.0.2,<6.0.0",
+    # "sphinx_rtd_theme>=0.2.4,<0.5",
+    # "autodocsumm>=0.1.10",
     # style check
     "flake8>=3.7.7",
     # "isort>=4.3.4",