fix types

huggingface · Aug 23, 2024 · 7d188ca · 7d188ca
1 parent 429ed38
commit 7d188ca
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/services/worker/tests/fixtures/files.py b/services/worker/tests/fixtures/files.py
@@ -131,7 +131,7 @@ def n_configs_paths(tmp_path_factory: pytest.TempPathFactory) -> list[str]:
     ]
     with open(readme, "w", newline="") as f:
         f.writelines(f"{line}\n" for line in lines)
-    files = [readme]
+    files = [str(readme)]
     for i in range(N):
         config_name = f"config{i}"
         path = directory / f"{config_name}.csv"
@@ -140,5 +140,5 @@ def n_configs_paths(tmp_path_factory: pytest.TempPathFactory) -> list[str]:
             writer.writeheader()
             for _ in range(1000):
                 writer.writerow({"text": config_name})
-        files.append(path)
+        files.append(str(path))
     return files
diff --git a/services/worker/tests/fixtures/hub.py b/services/worker/tests/fixtures/hub.py
@@ -255,7 +255,7 @@ def hub_public_big_no_info(datasets: Mapping[str, Dataset]) -> Iterator[str]:
 
 
 @pytest.fixture(scope="session")
-def hub_public_n_configs(n_configs_paths: str) -> Iterator[str]:
+def hub_public_n_configs(n_configs_paths: list[str]) -> Iterator[str]:
     repo_id = create_hub_dataset_repo(prefix="n-configs", file_paths=n_configs_paths)
     yield repo_id
     delete_hub_dataset_repo(repo_id=repo_id)