fix ruff workflow

kha-white · Jul 9, 2024 · 4b7a55c · 4b7a55c
1 parent cbfd3d4
commit 4b7a55c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/.github/workflows/ruff.yml b/.github/workflows/ruff.yml
@@ -4,5 +4,7 @@ jobs:
   ruff:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
-      - uses: chartboost/ruff-action@v1
+    - uses: actions/checkout@v4
+    - run: pip install --user ruff
+    - run: ruff check --output-format=github .
+    - run: ruff format --diff .
diff --git a/mokuro/manga_page_ocr.py b/mokuro/manga_page_ocr.py
@@ -37,7 +37,7 @@ def __init__(
 
         if not self.disable_ocr:
             cuda = torch.cuda.is_available()
-            device = 'cuda' if cuda and not force_cpu else 'cpu'
+            device = "cuda" if cuda and not force_cpu else "cpu"
             logger.info(f"Initializing text detector, using device {device}")
             self.text_detector = TextDetector(
                 model_path=cache.comic_text_detector, input_size=detector_input_size, device=device, act="leaky"