diff --git a/embodied/core/selectors.py b/embodied/core/selectors.py index 63eed6a2..fc65689d 100644 --- a/embodied/core/selectors.py +++ b/embodied/core/selectors.py @@ -211,6 +211,9 @@ def __init__(self, selectors, fractions, seed=0): self.fractions = np.array([fractions[key] for key in keys], np.float32) self.rng = np.random.default_rng(seed) + def __len__(self): + return len(self.selectors[0]) + def __call__(self): return self.rng.choice(self.selectors, p=self.fractions)()