Merge remote-tracking branch 'origin/main'

mir-group · Jun 27, 2024 · b591aa2 · b591aa2
2 parents f05b395 + 950b576
commit b591aa2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nequip/train/trainer.py b/nequip/train/trainer.py
@@ -1156,7 +1156,7 @@ def _parse_n_train_n_val(
             n = getattr(self, n_name)
             if isinstance(n, str) and "%" in n:
                 n_train_n_val.append(
-                    (float(n.strip("%")) / 100) * dataset_size
+                    (float(n.rstrip("%")) / 100) * dataset_size
                 )  # convert to float first
             elif isinstance(n, int):
                 n_train_n_val.append(n)