add support for benchmark.yaml (#415)

* add support for benchmark.yaml

recent zoo models use `benchmark.yaml` instead of `benchmarks.yaml`. adding this additional pathway so `benchmark.yaml` is downloaded in the bulk model download

* update files filter

* fix tests

---------

Co-authored-by: dbogunowicz <damian@neuralmagic.com>

src/sparsezoo/model/model.py

@@ -183,6 +183,9 @@ def __init__(self, source: str, download_path: Optional[str] = None):
         self.benchmarks: File = self._file_from_files(
             files, display_name="benchmarks.yaml"
         )
+        self.benchmark: File = self._file_from_files(
+            files, display_name="benchmark.yaml"
+        )
         self.eval_results: File = self._file_from_files(files, display_name="eval.yaml")
 
         # plaintext validation metrics optionally parsed from a zoo stub
@@ -243,6 +246,7 @@ def __init__(self, source: str, download_path: Optional[str] = None):
             "onnx_model": self.onnx_model,
             "analysis": self.analysis,
             "benchmarks": self.benchmarks,
+            "benchmark": self.benchmark,
             "eval_results": self.eval_results,
         }
 

src/sparsezoo/model/utils.py

@@ -58,6 +58,7 @@
     "benchmarking",
     "outputs",
     "onnx_gz",
+    "benchmark",
 }
 
 _LOGGER = logging.getLogger(__name__)

tests/sparsezoo/model/test_model.py

@@ -42,6 +42,7 @@
     "sample-labels",
     "sample-outputs",
     "benchmarks.yaml",
+    "benchmark.yaml",
     "eval.yaml",
     "analysis.yaml",
     "model.md",