[NFC] Fix few lint issues

Author: MrSidims

.github/workflows/ci.yml

@@ -14,6 +14,9 @@ on:
   schedule:
     - cron: '0 0 * * *'
 
+permissions:
+  contents: read
+
 jobs:
   test:
     runs-on: ubuntu-22.04

backend/server.py

@@ -1,34 +1,23 @@
-# FIXME: remove imports used on previous stages of development.
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-from pydantic import BaseModel
-from typing import List, Optional, Tuple
-
-import torch
-import torch.nn as nn
-from torch_mlir import fx
-from torch_mlir.compiler_utils import run_pipeline_with_repro_report
-from torch_mlir.fx import OutputType
-
 import subprocess
 import tempfile
-import triton
-import triton.language as tl
-import triton.compiler
-import ast
-import inspect
 import os
 import glob
 import uuid
 import hashlib
 import shutil
 import atexit
-import shutil
-
-import io
-import sys
+from typing import List, Optional, Tuple
 from contextlib import redirect_stdout, redirect_stderr
 
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+
+import torch
+import torch.nn as nn
+from torch_mlir import fx
+from torch_mlir.fx import OutputType
+
 app = FastAPI()
 
 app.add_middleware(

tests/test_torch_custom_pipeline.py

@@ -39,7 +39,7 @@ def test_torch_custom_pipeline(ir_type):
     output = response.json()["output"]
 
     assert "===== Initial IR =====" in output
-    assert "llvm.func @main(%arg0: !llvm.ptr, %arg1: !llvm.ptr, %arg2: i64, %arg3: i64, %arg4: i64, %arg5: i64, %arg6: i64)"
+    assert "llvm.func @main(%arg0: !llvm.ptr, %arg1: !llvm.ptr, %arg2: i64, %arg3: i64, %arg4: i64, %arg5: i64, %arg6: i64)" in output
     assert "===== IR after mlir-translate --mlir-to-llvmir =====" in output
     assert "define { ptr, ptr, i64, [2 x i64], [2 x i64] } @main" in output
     assert "===== IR after llc -mtriple=nvptx64 =====" in output