From 942f0e07acdd70b2838284d83fb75c838ab64c46 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Mon, 16 Dec 2024 20:42:20 +0100 Subject: [PATCH] ignore errors --- optimum_benchmark/backends/py_txi/backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum_benchmark/backends/py_txi/backend.py b/optimum_benchmark/backends/py_txi/backend.py index e7e298f4..3acd2e42 100644 --- a/optimum_benchmark/backends/py_txi/backend.py +++ b/optimum_benchmark/backends/py_txi/backend.py @@ -39,7 +39,7 @@ def load(self) -> None: try: self.tmpdir.cleanup() except Exception: - shutil.rmtree(self.tmpdir.name) + shutil.rmtree(self.tmpdir.name, ignore_errors=True) def download_pretrained_model(self) -> None: model_snapshot_folder = snapshot_download(self.config.model, **self.config.model_kwargs)