use correct zoo download dir param in onnx tests (#975)

* use correct zoo download dir param in onnx tests

* same change in src

Author: bfineran

src/sparseml/onnx/benchmark/info.py

@@ -290,9 +290,11 @@ def load_model(model: Any, **kwargs) -> ModelProto:
         raise ValueError("Model must not be None type")
 
     if isinstance(model, str) and model.startswith("zoo:"):
-        model = Model(model)
-        if "path" in kwargs:
-            model.path = kwargs["path"]
+        model = (
+            Model(model, download_path=kwargs["path"])
+            if "path" in kwargs
+            else Model(model)
+        )
 
     if isinstance(model, Model):
         # default to the main onnx file for the model

tests/sparseml/onnx/benchmark/test_info.py

@@ -37,8 +37,7 @@
 @pytest.fixture(scope="module")
 def mobilenet_fixture() -> Model:
     with tempfile.TemporaryDirectory() as onnx_dir:
-        model = Model(TEST_STUB)
-        model.path = onnx_dir
+        model = Model(TEST_STUB, download_path=onnx_dir)
         yield model