Merge branch 'main' into bnb/cond_obs_fwp

Author: bnb32

tests/data/config_disc_st_test.json

@@ -16,6 +16,10 @@
     {"alpha": 0.2, "class": "LeakyReLU"},
     {"class": "Conv3D", "filters": 256, "kernel_size": 3, "padding": "same", "strides": 2},
     {"alpha": 0.2, "class": "LeakyReLU"},
+    {"class": "Conv3D", "filters": 512, "kernel_size": 3, "padding": "same", "strides": 1},
+    {"alpha": 0.2, "class": "LeakyReLU"},
+    {"class": "Conv3D", "filters": 512, "kernel_size": 3, "padding": "same", "strides": 2},
+    {"alpha": 0.2, "class": "LeakyReLU"},
     {"class": "Flatten"},
     {"class": "Dense", "units": 2048},
     {"alpha": 0.2, "class": "LeakyReLU"},