Skip to content
Snippets Groups Projects
Verified Commit 080bce2f authored by Yoann Schneider's avatar Yoann Schneider :tennis:
Browse files

Revert "Only use files when computing model archive hash"

This reverts commit ceb2adbc.
parent 9caaa275
No related branches found
No related tags found
1 merge request!267Properly compute files hashes during models publication
Pipeline #79878 passed
...@@ -50,10 +50,12 @@ def create_archive(path: DirPath) -> Tuple[Path, Hash, FileSize, Hash]: ...@@ -50,10 +50,12 @@ def create_archive(path: DirPath) -> Tuple[Path, Hash, FileSize, Hash]:
# Create an uncompressed tar archive with all the needed files # Create an uncompressed tar archive with all the needed files
# Files hierarchy ifs kept in the archive. # Files hierarchy ifs kept in the archive.
file_list = []
with tarfile.open(path_to_tar_archive, "w") as tar: with tarfile.open(path_to_tar_archive, "w") as tar:
tar.add(path) for p in path.glob("**/*"):
file_list = [member for member in tar.getnames() if os.path.isfile(member)] x = p.relative_to(path)
tar.add(p, arcname=x, recursive=False)
file_list.append(p)
# Sort by path # Sort by path
file_list.sort() file_list.sort()
......
...@@ -26,7 +26,7 @@ from arkindex_worker.worker import BaseWorker, ElementsWorker ...@@ -26,7 +26,7 @@ from arkindex_worker.worker import BaseWorker, ElementsWorker
from arkindex_worker.worker.transcription import TextOrientation from arkindex_worker.worker.transcription import TextOrientation
FIXTURES_DIR = Path(__file__).resolve().parent / "data" FIXTURES_DIR = Path(__file__).resolve().parent / "data"
SAMPLES_DIR = Path("tests") / "samples" SAMPLES_DIR = Path(__file__).resolve().parent / "samples"
__yaml_cache = {} __yaml_cache = {}
...@@ -280,11 +280,6 @@ def model_file_dir(): ...@@ -280,11 +280,6 @@ def model_file_dir():
return SAMPLES_DIR / "model_files" return SAMPLES_DIR / "model_files"
@pytest.fixture
def model_file_dir_with_subfolder():
return SAMPLES_DIR / "model_files_with_subfolder"
@pytest.fixture @pytest.fixture
def fake_dummy_worker(): def fake_dummy_worker():
api_client = MockApiClient() api_client = MockApiClient()
......
Wow this is actually the data of the best model ever created on Arkindex
\ No newline at end of file
Wow this is actually the data of the best model ever created on Arkindex
\ No newline at end of file
...@@ -45,24 +45,6 @@ def test_create_archive(model_file_dir): ...@@ -45,24 +45,6 @@ def test_create_archive(model_file_dir):
assert not os.path.exists(zst_archive_path), "Auto removal failed" assert not os.path.exists(zst_archive_path), "Auto removal failed"
def test_create_archive_with_subfolder(model_file_dir_with_subfolder):
"""Create an archive when the model's file is in a folder containing a subfolder"""
with create_archive(path=model_file_dir_with_subfolder) as (
zst_archive_path,
hash,
size,
archive_hash,
):
assert os.path.exists(zst_archive_path), "The archive was not created"
assert (
hash == "e2fa86cefc33b24502ad4151a638dd29"
), "Hash was not properly computed"
assert 300 < size < 1500
assert not os.path.exists(zst_archive_path), "Auto removal failed"
@pytest.mark.parametrize( @pytest.mark.parametrize(
"tag, description", "tag, description",
[ [
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment