diff --git a/requirements.txt b/requirements.txt index ac307a31f50855cb27ba240efac6ea3258004a61..ea408dfadec1ff8f7170f95fb468bdd808f6fcd6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -arkindex-base-worker==0.3.2 +arkindex-base-worker==0.3.3-rc2 arkindex-export==0.1.2 imageio==2.27.0 opencv-python-headless==4.7.0.72 diff --git a/worker_generic_training_dataset/worker.py b/worker_generic_training_dataset/worker.py index 78ba703307d038d7ab9e77e8094a62cc9a07bf79..1bc7bd5ad3d0929670db9db37dce99fbaf2f75f9 100644 --- a/worker_generic_training_dataset/worker.py +++ b/worker_generic_training_dataset/worker.py @@ -96,7 +96,7 @@ class DatasetExtractor(BaseWorker): database_path = ( Path(self.args.database) if self.is_read_only - else self.workdir / "db.sqlite" + else self.work_dir / "db.sqlite" ) if database_path.exists(): database_path.unlink() @@ -276,7 +276,7 @@ class DatasetExtractor(BaseWorker): self.process_split(split_name, split_id, image_folder) # TAR + ZSTD Image folder and store as task artifact - zstd_archive_path = Path(self.work_dir) / "arkindex_data.zstd" + zstd_archive_path = self.work_dir / "arkindex_data.zstd" logger.info(f"Compressing the images to {zstd_archive_path}") create_tar_zstd_archive(folder_path=image_folder, destination=zstd_archive_path)