Upload test_multi_lora.py with huggingface_hub

test_multi_lora.py

@@ -236,13 +236,30 @@ def main():
     print("\n" + "=" * 60)
     print("Test Summary")
     print("=" * 60)
-    print(json.dumps(results, indent=2, default=str, ensure_ascii=False))
+    summary = json.dumps(results, indent=2, default=str, ensure_ascii=False)
+    print(summary)
 
-    # Exit code
-    failed = any(
-        r.get("status") == "FAIL" for r in results.values() if isinstance(r, dict)
-    )
-    sys.exit(1 if failed else 0)
+    # 결과를 HF repo에 업로드
+    try:
+        from huggingface_hub import HfApi
+        result_path = "/tmp/test_results.json"
+        with open(result_path, "w") as f:
+            f.write(summary)
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj=result_path,
+            path_in_repo="test_results.json",
+            repo_id="umyunsang/govon-multi-lora-test",
+            repo_type="space",
+        )
+        print("\nResults uploaded to repo as test_results.json")
+    except Exception as e:
+        print(f"\nFailed to upload results: {e}")
+
+    # 종료하지 않고 대기 (Space가 RUNNING 상태 유지)
+    print("\nTest complete. Keeping container alive...")
+    import signal
+    signal.pause()
 
 
 if __name__ == "__main__":