From 01987b21def0f3cd9ab31fd82cd3777336609611 Mon Sep 17 00:00:00 2001 From: Damian Date: Tue, 21 Nov 2023 10:21:36 +0000 Subject: [PATCH] initial commit --- tests/sparseml/transformers/utils/test_helpers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/sparseml/transformers/utils/test_helpers.py b/tests/sparseml/transformers/utils/test_helpers.py index ae61578ccfe..ac668bdadc0 100644 --- a/tests/sparseml/transformers/utils/test_helpers.py +++ b/tests/sparseml/transformers/utils/test_helpers.py @@ -32,6 +32,7 @@ def test_save_zoo_directory(stub, tmp_path_factory): zoo_model = Model(stub, path_to_training_outputs) zoo_model.download() + zoo_model.deployment_tar.unzip() zoo_model.sample_inputs.unzip() zoo_model.sample_outputs["framework"].unzip()