a1c00l commited on
Commit
475172d
·
verified ·
1 Parent(s): ae6cf20

Update src/aibom_generator/generator.py

Browse files
Files changed (1) hide show
  1. src/aibom_generator/generator.py +1 -1
src/aibom_generator/generator.py CHANGED
@@ -120,7 +120,7 @@ class AIBOMGenerator:
120
  metadata["ai:framework"] = "PyTorch" if "transformers" in metadata.get("library_name", "") else "Unknown"
121
 
122
 
123
- return {k: v for k, v in metadata.items() if v is not None}
124
 
125
  def _extract_unstructured_metadata(self, model_card: ModelCard) -> Dict[str, Any]:
126
  return {}
 
120
  metadata["ai:framework"] = "PyTorch" if "transformers" in metadata.get("library_name", "") else "Unknown"
121
 
122
 
123
+ return {k: v for k, v in metadata.items() if v is not None}
124
 
125
  def _extract_unstructured_metadata(self, model_card: ModelCard) -> Dict[str, Any]:
126
  return {}