Refactor label generation to use an external script for improved maintainability #38

Author: ziagham

.github/workflows/release.yml

@@ -93,19 +93,8 @@ jobs:
             # copy metadata
             cp "$ONNX_DIR/metadata.json" release_temp/ 2>/dev/null || true
 
-            # Generate labels.json from config
-            python -c "
-            import json, os, sys
-            onnx_dir = sys.argv[1]
-            if not onnx_dir:
-                sys.exit(1)
-            config_path = os.path.join(onnx_dir, 'config.json')
-            with open(config_path) as f:
-                config = json.load(f)
-            labels = [config['id2label'][str(i)] for i in range(len(config['id2label']))]
-            with open('release_temp/labels.json', 'w') as f:
-                json.dump(labels, f)
-            " "$ONNX_DIR"
+            # Generate labels.json from config (using an external script)
+            python src/generate_labels_json.py "$ONNX_DIR"
 
             # Zip the release
             (cd release_temp && zip -r "../release_artifacts/$ZIP_NAME" .)

src/generate_labels_json.py

@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+"""Generate labels.json from an ONNX export directory (config.json)."""
+
+import json
+import os
+import sys
+
+def main():
+    if len(sys.argv) != 2:
+        print("Usage: generate_labels_json.py <onnx_dir>")
+        sys.exit(1)
+
+    onnx_dir = sys.argv[1]
+    config_path = os.path.join(onnx_dir, "config.json")
+
+    with open(config_path, "r") as f:
+        config = json.load(f)
+
+    # Map id2label to an ordered list of labels
+    labels = [config["id2label"][str(i)] for i in range(len(config["id2label"]))]
+
+    # Output to release_temp/labels.json (working directory is repo root)
+    os.makedirs("release_temp", exist_ok=True)
+    with open("release_temp/labels.json", "w") as f:
+        json.dump(labels, f)
+
+    print(f"labels.json written with {len(labels)} labels.")
+
+if __name__ == "__main__":
+    main()