fixes to split slice (#97)

the tutorials now use the new `split/slice` API of `metatensor`
lab-cosmo · Nov 13, 2024 · 5f55108 · 5f55108
1 parent bc91abb
commit 5f55108
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/examples/lode-linear/lode-linear.py b/examples/lode-linear/lode-linear.py
@@ -284,14 +284,14 @@
 # That we use as input to the :py:func:`metatensor.split()` function
 
 X_sr_train, X_sr_test = metatensor.split(
-    X_sr, axis="samples", grouped_labels=grouped_labels
+    X_sr, axis="samples", selections=grouped_labels
 )
 
 X_lr_train, X_lr_test = metatensor.split(
-    X_lr, axis="samples", grouped_labels=grouped_labels
+    X_lr, axis="samples", selections=grouped_labels
 )
 
-y_train, y_test = metatensor.split(y, axis="samples", grouped_labels=grouped_labels)
+y_train, y_test = metatensor.split(y, axis="samples", selections=grouped_labels)
 
 
 # %%

diff --git a/examples/sample-selection/sample-selection.py b/examples/sample-selection/sample-selection.py
@@ -313,7 +313,7 @@
 atom_soap_single_block_fps = metatensor.slice(
     atom_soap_single_block,
     axis="properties",
-    labels=feat_fps.support.block(0).properties,
+    selection=feat_fps.support.block(0).properties,
 )
 
 print(