From cc75c26d1340ddf230c441672286cbe9e759e696 Mon Sep 17 00:00:00 2001
From: Sukhil Patel <sukhil@Sukhils-MacBook-Pro.local>
Date: Tue, 19 Mar 2024 13:58:21 +0000
Subject: [PATCH] Remove references of removed datamodule config parameters

---
 configs.example/experiment/baseline.yaml       | 4 ----
 configs.example/experiment/conv3d_sat_nwp.yaml | 4 ----
 configs.example/experiment/example_simple.yaml | 4 +---
 3 files changed, 1 insertion(+), 11 deletions(-)

diff --git a/configs.example/experiment/baseline.yaml b/configs.example/experiment/baseline.yaml
index 0f5b9cf9..f5c87e98 100644
--- a/configs.example/experiment/baseline.yaml
+++ b/configs.example/experiment/baseline.yaml
@@ -19,7 +19,3 @@ validate_only: "1" # by putting this key in the config file, the model does not
 trainer:
   min_epochs: 1
   max_epochs: 1
-
-datamodule:
-  n_train_data: 2
-  n_val_data: 10
diff --git a/configs.example/experiment/conv3d_sat_nwp.yaml b/configs.example/experiment/conv3d_sat_nwp.yaml
index d4de2a71..26b951f2 100644
--- a/configs.example/experiment/conv3d_sat_nwp.yaml
+++ b/configs.example/experiment/conv3d_sat_nwp.yaml
@@ -19,9 +19,5 @@ trainer:
   min_epochs: 1
   max_epochs: 10
 
-datamodule:
-  n_train_data: 4000
-  n_val_data: 400
-
 model:
   conv3d_channels: 32
diff --git a/configs.example/experiment/example_simple.yaml b/configs.example/experiment/example_simple.yaml
index efcc4eae..0cdcda08 100644
--- a/configs.example/experiment/example_simple.yaml
+++ b/configs.example/experiment/example_simple.yaml
@@ -22,8 +22,6 @@ trainer:
   max_epochs: 2
 
 datamodule:
-  n_train_data: 2
-  n_val_data: 2
-  fake_data: 1
+  batch_size: 32
 
 validate_only: "1" # by putting this key in the config file, the model does not get trained.