Add assertion

ludwig-ai · Jun 2, 2022 · 7a3db8c · 7a3db8c
1 parent 849ccbf
commit 7a3db8c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ludwig/data/dataset/ray.py b/ludwig/data/dataset/ray.py
@@ -96,6 +96,8 @@ def pipeline(
         if not enable_windowing:
             pipe = self.ds.repeat()
         else:
+            if window_size_bytes is None:
+                raise ValueError("window_size_bytes must not be None when enable_windowing is True.")
             pipe = self.ds.window(bytes_per_window=window_size_bytes).repeat()
         if shuffle:
             pipe = pipe.random_shuffle_each_window()