Add missing Pyre mode headers] [batch:21/1522] [shard:38/N]

Differential Revision: D68144453

examples/multi_thread_preprocessing.py

@@ -5,6 +5,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 """This example shows how to run PyTorch tarnsform in SPDL Pipeline,
 and compares its performance against PyTorch DataLoader.
 

src/spdl/io/_zip.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from collections.abc import Mapping
 
 import spdl.lib

src/spdl/source/_type.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 __all__ = ["IterableWithShuffle"]
 
 from collections.abc import Iterator

tests/spdl_unittest/conftest.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import subprocess
 from dataclasses import dataclass
 

tests/spdl_unittest/dataloader/dataloader_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import time
 
 from spdl.dataloader import DataLoader

tests/spdl_unittest/dataloader/iterator_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from collections.abc import Iterator
 from unittest.mock import patch
 

tests/spdl_unittest/dataloader/pipeline_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import asyncio
 import random
 import threading

tests/spdl_unittest/dataloader/source_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from collections.abc import Iterable
 from pathlib import Path
 from tempfile import TemporaryDirectory

tests/spdl_unittest/io/async_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import time
 
 import numpy as np

tests/spdl_unittest/io/buffer_conversion_refcount_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import gc
 import sys
 

tests/spdl_unittest/io/concurrent_audio_decoding_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 
 import numpy as np
 import pytest

tests/spdl_unittest/io/concurrent_image_decoding_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import numpy as np
 import pytest
 import spdl.io

tests/spdl_unittest/io/configs_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import pytest
 import spdl.io
 

tests/spdl_unittest/io/demuxer_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import numpy as np
 import pytest
 import spdl.io

tests/spdl_unittest/io/encoding_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 from itertools import product
 from tempfile import NamedTemporaryFile
 

tests/spdl_unittest/io/frames_clone_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import numpy as np
 import pytest
 import spdl.io

tests/spdl_unittest/io/frames_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import spdl.io
 
 

tests/spdl_unittest/io/packets_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import time
 
 import numpy as np

tests/spdl_unittest/io/streaming_video_decoding_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import numpy as np
 import pytest
 import spdl.io

tests/spdl_unittest/io/video_frame_slice_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import numpy as np
 import pytest
 import spdl.io

tests/spdl_unittest/io/zero_copy_bytes_passing_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import numpy as np
 import pytest
 import spdl.io

tests/spdl_unittest/io/zip_test.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
+
 import tempfile
 
 import numpy as np