replace update_issue_comment with dedicated workflow steps

Author: j2kun

.github/workflows/torch_mlir_coverage.yml

@@ -44,8 +44,17 @@ jobs:
           name: failure-logs
           path: scripts/torch_ci/failure_logs/
 
-      - name: Update Issue Comment
-        env:
-          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-        run: |
-          python scripts/torch_ci/update_issue_comment.py coverage_table.md
+      - name: Find Coverage Comment
+        id: fc
+        uses: peter-evans/find-comment@b30e6a3c0ed37e7c023ccd3f1db5c6c0b0c23aad
+        with:
+          issue-number: 2923
+          body-includes: '<!-- TORCH_COVERAGE_TABLE -->'
+
+      - name: Create or Update Coverage Comment
+        uses: peter-evans/create-or-update-comment@e8674b075228eee787fea43ef493e45ece1004c9
+        with:
+          comment-id: ${{ steps.fc.outputs.comment-id }}
+          issue-number: 2923
+          body-path: coverage_table.md
+          edit-mode: replace

scripts/torch_ci/BUILD

@@ -18,16 +18,3 @@ py_test(
         ":entrypoint_lib",
     ],
 )
-
-py_library(
-    name = "update_issue_comment_lib",
-    srcs = ["update_issue_comment.py"],
-)
-
-py_test(
-    name = "test_update_issue_comment",
-    srcs = ["test_update_issue_comment.py"],
-    deps = [
-        ":update_issue_comment_lib",
-    ],
-)

scripts/torch_ci/entrypoint.py

@@ -68,7 +68,7 @@ def run_all(self):
     models = self.find_models()
     results = {}
     for model in sorted(models):
-      print(f"Running coverage for {model}...")
+      print(f"Running coverage for {model}...", file=sys.stderr)
       export_result = self.run_export(model)
       if export_result.returncode != 0:
         results[model] = ("FAIL (Export)", export_result.stderr)
@@ -77,7 +77,13 @@ def run_all(self):
 
       heir_opt_result = self.run_heir_opt(model)
       if heir_opt_result is None:
-        results[model] = ("FAIL (No MLIR)", "model.mlir not found")
+        results[model] = (
+            "FAIL (No MLIR)",
+            "torch-mlir failed to produce model.mlir",
+        )
+        self.save_failure_log(
+            model, "heir-opt", "torch-mlir failed to produce model.mlir"
+        )
         continue
 
       if heir_opt_result.returncode != 0:
@@ -96,7 +102,8 @@ def save_failure_log(self, model_name, stage, content):
       f.write(content)
 
   def format_results_table(self, results):
-    table = "| Operator | Status | Details |\n"
+    table = "<!-- TORCH_COVERAGE_TABLE -->\n\n"
+    table += "| Operator | Status | Details |\n"
     table += "| --- | --- | --- |\n"
     for model, (status, details) in results.items():
       details_str = details.replace("\n", " ")[:100] if details else ""