[Fix] Fix the sparsezoo.Model to "see" metrics.yaml

src/sparsezoo/model/model.py

@@ -188,6 +188,8 @@ def __init__(self, source: str, download_path: Optional[str] = None):
  )
  self.eval_results: File = self._file_from_files(files, display_name="eval.yaml")
 
+ self.metrics: File = self._file_from_files(files, display_name="metrics.yaml")
+
  # plaintext validation metrics optionally parsed from a zoo stub
  self.validation_results: Optional[
  Dict[str, List[ModelResult]]
@@ -248,6 +250,7 @@ def __init__(self, source: str, download_path: Optional[str] = None):
  "benchmarks": self.benchmarks,
  "benchmark": self.benchmark,
  "eval_results": self.eval_results,
+ "metrics": self.metrics,
  }
 
  self.inference_runner = InferenceRunner(

src/sparsezoo/model/utils.py

@@ -59,6 +59,7 @@
  "outputs",
  "onnx_gz",
  "benchmark",
+ "metrics",
 }
 
 _LOGGER = logging.getLogger(__name__)
@@ -345,6 +346,7 @@ def restructure_request_json(
  that will not be filtered out during restructuring
  :return: restructured files
  """
+
  # create `training` folder
  training_dicts_list = fetch_from_request_json(
  request_json, "file_type", "framework"

tests/sparsezoo/model/test_model.py

@@ -46,6 +46,7 @@
  "eval.yaml",
  "analysis.yaml",
  "model.md",
+ "metrics.yaml",
 }
 
 files_nlp = copy.copy(files_ic)