[Cherry-Pick][Fix] Failing test_save_zoo_directory tests (#1869)

* "AttributeError: 'torch._C.Value' object has no attribute 'float'" sparseml.transformers.export_onnx for zoo:bert-base_cased-squad_wikipedia_bookcorpus-pruned80.4block_quantized (#1847) * [Fix] Failing tests
neuralmagic · Nov 30, 2023 · aac6b90 · aac6b90
1 parent c052e27
commit aac6b90
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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()