Skip to content

Commit 3985a37

Browse files
committed
hf stuff + pytest warnings
1 parent 86acbaf commit 3985a37

3 files changed

Lines changed: 0 additions & 4 deletions

File tree

stopes/pipelines/tests/test_global_mining.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,6 @@ def encode_to_np(
129129
(False, True, True, "speech"),
130130
],
131131
)
132-
@pytest.mark.asyncio(scope="session")
133132
def test_global_mining_pipeline(
134133
tmp_path: Path, split_langs: bool, use_meta: bool, fp16: bool, modality: str
135134
) -> None:

stopes/utils/sharding/hf_shards.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,6 @@ def __enter__(self):
103103
cache_dir=_cache_dir,
104104
download_mode=_download_mode,
105105
split=self.split,
106-
trust_remote_code=self.trust_remote_code,
107106
)
108107
if self.split is None: # _data is a DatasetDict, convert to Dataset
109108
_data = concatenate_datasets(

stopes/utils/test_hf_shards.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ def test_shard_iteration():
1818
split="test",
1919
index=0,
2020
num_shards=50,
21-
trust_remote_code=True,
2221
)
2322
with shard:
2423
item = next(iter(shard))
@@ -37,7 +36,6 @@ def test_input_config():
3736
input_file="Fraser/mnist-text-small",
3837
split="test",
3938
num_shards=50,
40-
trust_remote_code=True,
4139
)
4240
shards = input_config.make_shards()
4341
first_shard = shards[0]

0 commit comments

Comments
 (0)