add max_size inside ShimmerDataset for DataDomain

ruflab · Oct 15, 2024 · d9dea96 · d9dea96
1 parent 6556d02
commit d9dea96
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/shimmer/data/dataset.py b/shimmer/data/dataset.py
@@ -87,6 +87,7 @@ def __init__(
             self.domains[domain.kind] = domain_cls(
                 dataset_path,
                 split,
+                max_size,
                 transform,
                 self.domain_args.get(domain.kind, None),
             )