small fix to legacy tutorial

FAIR-Chem · Jul 11, 2024 · 152565b · 152565b
1 parent 1e6e101
commit 152565b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/docs/legacy_tutorials/OCP_Tutorial.md b/docs/legacy_tutorials/OCP_Tutorial.md
@@ -1427,7 +1427,6 @@ task = {
       'grad_input': 'atomic forces',
       'train_on_free_atoms': True,
       'eval_on_free_atoms': True,
-      'relax_dataset': {"src": relax_dataset},
       'write_pos': True,
       'relaxation_steps': 200,
       'num_relaxation_batches': 1,
@@ -1489,10 +1488,11 @@ optimizer = {
     'force_coefficient': 100,
 }
 # Dataset
-dataset = [
-  {'src': train_src, 'normalize_labels': False}, # train set
-  {'src': val_src}, # val set (optional)
-]
+dataset = {
+  'dataset': {'src': train_src, 'normalize_labels': False}, # train set
+  'val_dataset': {'src': val_src}, # val set (optional)
+  'relax_dataset': {"src": relax_dataset},
+}
 ```