Update

.pre-commit-config.yaml

@@ -14,13 +14,13 @@ repos:
       - id: requirements-txt-fixer
       - id: trailing-whitespace
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.8.6
+    rev: v0.11.10
     hooks:
       - id: ruff
         args: ["--fix"]
       - id: ruff-format
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.14.1
+    rev: v1.15.0
     hooks:
       - id: mypy
         args: ["--ignore-missing-imports"]

model.py

@@ -9,7 +9,7 @@ from diffusers.utils import export_to_ply
 
 
 class Model:
-    def __init__(self):
+    def __init__(self) -> None:
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.pipe = ShapEPipeline.from_pretrained("openai/shap-e", torch_dtype=torch.float16)
         self.pipe.to(self.device)
@@ -23,9 +23,9 @@ class Model:
         mesh = mesh.apply_transform(rot)
         rot = trimesh.transformations.rotation_matrix(np.pi, [0, 1, 0])
         mesh = mesh.apply_transform(rot)
-        mesh_path = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
-        mesh.export(mesh_path.name, file_type="glb")
-        return mesh_path.name
+        with tempfile.NamedTemporaryFile(suffix=".glb", delete=False) as mesh_path:
+            mesh.export(mesh_path.name, file_type="glb")
+            return mesh_path.name
 
     def run_text(self, prompt: str, seed: int = 0, guidance_scale: float = 15.0, num_steps: int = 64) -> str:
         generator = torch.Generator(device=self.device).manual_seed(seed)
@@ -36,9 +36,9 @@ class Model:
             num_inference_steps=num_steps,
             output_type="mesh",
         ).images
-        ply_path = tempfile.NamedTemporaryFile(suffix=".ply", delete=False, mode="w+b")
-        export_to_ply(images[0], ply_path.name)
-        return self.to_glb(ply_path.name)
+        with tempfile.NamedTemporaryFile(suffix=".ply", delete=False, mode="w+b") as ply_path:
+            export_to_ply(images[0], ply_path.name)
+            return self.to_glb(ply_path.name)
 
     def run_image(
         self, image: PIL.Image.Image, seed: int = 0, guidance_scale: float = 3.0, num_steps: int = 64
@@ -51,6 +51,6 @@ class Model:
             num_inference_steps=num_steps,
             output_type="mesh",
         ).images
-        ply_path = tempfile.NamedTemporaryFile(suffix=".ply", delete=False, mode="w+b")
-        export_to_ply(images[0], ply_path.name)
-        return self.to_glb(ply_path.name)
+        with tempfile.NamedTemporaryFile(suffix=".ply", delete=False, mode="w+b") as ply_path:
+            export_to_ply(images[0], ply_path.name)
+            return self.to_glb(ply_path.name)

pyproject.toml

@@ -29,8 +29,7 @@ ignore = [
     "D213",   # multi-line-summary-second-line
     "E501",   # line-too-long
     "SIM117", # multiple-with-statements
-]
-extend-ignore = [
+    #
     "D100",    # undocumented-public-module
     "D101",    # undocumented-public-class
     "D102",    # undocumented-public-method
@@ -51,5 +50,11 @@ unfixable = [
     "F401", # unused-import
 ]
 
+[tool.ruff.lint.pydocstyle]
+convention = "google"
+
+[tool.ruff.lint.per-file-ignores]
+"*.ipynb" = ["T201", "T203"]
+
 [tool.ruff.format]
 docstring-code-format = true

utils.py

@@ -5,5 +5,5 @@ from settings import MAX_SEED
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
+        seed = random.randint(0, MAX_SEED)  # noqa: S311
     return seed