ruff check --fix

Author: jmaczan

perf_test.py

@@ -1,6 +1,5 @@
 """Test WebGPU buffer creation overhead."""
 import torch
-import torch_webgpu
 import time
 
 device = torch.device("webgpu")

pyproject.toml

@@ -27,7 +27,7 @@ dependencies = [
 
 [project.optional-dependencies]
 docs = ["mkdocs", "mkdocs-material"]
-dev = ["pytest", "mkdocs", "mkdocs-material"]
+dev = ["pytest", "mkdocs", "mkdocs-material", "ruff"]
 
 [project.urls]
 Homepage = "https://github.com/jmaczan/torch-webgpu"

python/torch_webgpu/compiler/low_ir.py

@@ -4,7 +4,7 @@
 
 from .compiler_pass import CompilerPass
 from .ir import IRNode
-from .high_ir import HighIRCreateTensor, HighIRNode, HighIROp, HighIRArange
+from .high_ir import HighIRCreateTensor, HighIRNode, HighIROp
 
 
 class LowIROp(StrEnum):

python/torch_webgpu/compiler/lowering.py

@@ -1,4 +1,4 @@
-from typing import Any, Callable, List
+from typing import Callable, List
 from functools import partial
 from .low_ir import (
     LowIRCreateBuffer,
@@ -9,7 +9,6 @@
     LowIRRunShader,
     LowIRWriteBuffer,
 )
-import torch_webgpu
 import torch
 
 Runtime = dict

setup.py

@@ -4,7 +4,6 @@
 import shutil
 from pathlib import Path
 from setuptools import setup, find_packages
-from setuptools.command.build_ext import build_ext as _build_ext
 from torch.utils.cpp_extension import CppExtension, BuildExtension
 
 # Try to load .env for local development
@@ -128,7 +127,7 @@ def run(self):
 
 if system == "Linux":
     extra_compile_args = ["-std=c++17", "-O2"]
-    extra_link_args = [f"-Wl,-rpath,$ORIGIN/libs"]
+    extra_link_args = ["-Wl,-rpath,$ORIGIN/libs"]
 elif system == "Darwin":
     extra_compile_args = ["-std=c++17", "-O2"]
     extra_link_args = ["-Wl,-rpath,@loader_path/libs"]

tests/compile/test_compile_ops.py

@@ -1,7 +1,6 @@
 """Tests for torch.compile with webgpu backend on individual ops."""
 import pytest
 import torch
-import torch_webgpu
 from torch_webgpu import webgpu_backend
 
 

tests/compile/test_high_ir.py

@@ -1,5 +1,4 @@
 """Tests for High IR op mappings and node creation."""
-import pytest
 import torch
 import torch.nn.functional as F
 import operator

tests/compile/test_low_ir.py

@@ -1,5 +1,4 @@
 """Tests for Low IR op mappings."""
-import pytest
 
 from torch_webgpu.compiler.high_ir import HighIROp
 from torch_webgpu.compiler.low_ir import (

tests/test_compile_dev.py

@@ -1,5 +1,4 @@
 import torch
-import torch_webgpu
 from torch_webgpu import webgpu_backend
 
 

tests/test_dev.py

@@ -1,5 +1,4 @@
 import torch
-import torch_webgpu
 from torch_webgpu import webgpu_backend