From 5d55b19cbd90e2c9921039c7ced75bf1ef2f7b2b Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Wed, 5 Jun 2024 16:47:52 +0000 Subject: [PATCH] Fix tests --- lerobot/common/datasets/transforms.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/lerobot/common/datasets/transforms.py b/lerobot/common/datasets/transforms.py index cf1baced..25c68227 100644 --- 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):