Fix tests

2024-06-05 16:47:52 +00:00 · 2024-06-05 16:47:52 +00:00 · 5d55b19cbd
parent 641d349df4
commit 5d55b19cbd
1 changed files with 6 additions and 4 deletions
--- a/lerobot/common/datasets/transforms.py
+++ b/lerobot/common/datasets/transforms.py
@ -66,10 +66,12 @@ class RandomSubsetApply(Transform):
        return outputs

    def extra_repr(self) -> str:
-        format_string = [f"N={self.N}"]
-        for t in self.transforms:
-            format_string.append(f"    {t}")
-        return "\n".join(format_string)
+        return (
+            f"transforms={self.transforms}, "
+            f"p={self.p}, "
+            f"n_subset={self.n_subset}, "
+            f"random_order={self.random_order}"
+        )


 class RangeRandomSharpness(Transform):