Skip to content

Commit fd942ee

Browse files
pmeierfacebook-github-bot
authored andcommitted
fix deprecation warning for datapipes (#324)
Summary: Sister PR to pytorch/pytorch#74685. Fixes #322. CI will fail until the change is merged upstream. Pull Request resolved: #324 Reviewed By: ejguan Differential Revision: D35549848 Pulled By: NivekT fbshipit-source-id: 143f0fd9e656bd93ea540c985105b8c0d014c50f
1 parent 9ad73d8 commit fd942ee

File tree

3 files changed

+24
-3
lines changed

3 files changed

+24
-3
lines changed

torchdata/datapipes/iter/util/tararchiveloader.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,5 +85,12 @@ class TarArchiveReaderIterDataPipe(IterDataPipe[Tuple[str, BufferedIOBase]]):
8585
"""
8686

8787
def __new__(cls, datapipe: Iterable[Tuple[str, BufferedIOBase]], mode: str = "r:*", length: int = -1):
88-
deprecation_warning(type(cls).__name__, new_name="TarArchiveLoader")
88+
deprecation_warning(
89+
cls.__name__,
90+
deprecation_version="0.4",
91+
removal_version="0.6",
92+
old_functional_name="read_from_tar",
93+
new_class_name="TarArchiveLoader",
94+
new_functional_name="load_from_tar",
95+
)
8996
return TarArchiveLoaderIterDataPipe(datapipe, mode, length)

torchdata/datapipes/iter/util/xzfileloader.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,5 +74,12 @@ class XzFileReaderIterDataPipe(IterDataPipe[Tuple[str, BufferedIOBase]]):
7474
"""
7575

7676
def __new__(cls, datapipe: Iterable[Tuple[str, BufferedIOBase]], length: int = -1):
77-
deprecation_warning(type(cls).__name__, new_name="XzFileLoader")
77+
deprecation_warning(
78+
cls.__name__,
79+
deprecation_version="0.4",
80+
removal_version="0.6",
81+
old_functional_name="read_from_xz",
82+
new_class_name="XzFileLoader",
83+
new_functional_name="load_from_xz",
84+
)
7885
return XzFileLoaderIterDataPipe(datapipe, length)

torchdata/datapipes/iter/util/ziparchiveloader.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,5 +86,12 @@ class ZipArchiveReaderIterDataPipe(IterDataPipe[Tuple[str, BufferedIOBase]]):
8686
"""
8787

8888
def __new__(cls, datapipe: Iterable[Tuple[str, BufferedIOBase]], length: int = -1):
89-
deprecation_warning(type(cls).__name__, new_name="ZipArchiveLoader")
89+
deprecation_warning(
90+
cls.__name__,
91+
deprecation_version="0.4",
92+
removal_version="0.6",
93+
old_functional_name="read_from_zip",
94+
new_class_name="ZipArchiveLoader",
95+
new_functional_name="load_from_zip",
96+
)
9097
return ZipArchiveLoaderIterDataPipe(datapipe, length)

0 commit comments

Comments
 (0)