Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
AlexanderDokuchaev committed Nov 4, 2024
1 parent a812a27 commit d51405d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
anomalib==0.6.0
openvino==2024.4
setuptools<=72.1.0
numpy<2
5 changes: 3 additions & 2 deletions tests/cross_fw/examples/run_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -250,8 +250,9 @@ def quantization_aware_training_torch_anomalib(data: Union[str, None]):
if data is not None:
dataset_path.mkdir(parents=True, exist_ok=True)
tar_file_path = Path(data) / mvtec.DOWNLOAD_INFO.url.split("/")[-1]
with tarfile.open(tar_file_path) as tar_file:
tar_file.extractall(dataset_path)
if not tar_file_path.exists():
with tarfile.open(tar_file_path) as tar_file:
tar_file.extractall(dataset_path)

# Set manual seed and determenistic cuda mode to make the test determenistic
set_torch_cuda_seed()
Expand Down

0 comments on commit d51405d

Please sign in to comment.