updated tests

pytorch · Mar 22, 2022 · 83db89d · 83db89d
1 parent ac4e4c8
commit 83db89d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/test_serialization.py b/test/test_serialization.py
@@ -139,6 +139,7 @@ def _serialization_test_for_dp_with_children(self, dp1, dp2):
  def test_serializable(self):
  picklable_datapipes: List = [
  (iterdp.BucketBatcher, IterableWrapper([0, 0, 0, 0, 0, 0, 0]), (5,), {}),
+ (iterdp.Bz2FileLoader, None, (), {}),
  (
  iterdp.CSVDictParser,
  IterableWrapper(
@@ -256,6 +257,7 @@ def test_serializable(self):
  # Most of them return streams not comparable by `self.assertEqual`
  # Others are similar to caching where the outputs depend on other DataPipes
  dp_skip_comparison = {
+ iterdp.Bz2FileLoader,
  iterdp.Decompressor,
  iterdp.FileOpener,
  iterdp.FSSpecFileOpener,