Skip to content

Commit aac6b90

Browse files
authored
[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
1 parent c052e27 commit aac6b90

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/sparseml/transformers/utils/test_helpers.py

+1
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ def test_save_zoo_directory(stub, tmp_path_factory):
3232
zoo_model = Model(stub, path_to_training_outputs)
3333
zoo_model.download()
3434

35+
zoo_model.deployment_tar.unzip()
3536
zoo_model.sample_inputs.unzip()
3637
zoo_model.sample_outputs["framework"].unzip()
3738

0 commit comments

Comments
 (0)