From d9dea9669b8a10e55d880c7a6d8b0f6c2293507f Mon Sep 17 00:00:00 2001
From: leopoldmt <leopold.maytie@gmail.com>
Date: Tue, 15 Oct 2024 13:41:17 +0000
Subject: [PATCH] add max_size inside ShimmerDataset for DataDomain

---
 shimmer/data/dataset.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/shimmer/data/dataset.py b/shimmer/data/dataset.py
index 8776ccc..6e3b19f 100644
--- 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),
             )