fix mps check

ilastik · Jun 28, 2023 · 6e91b76 · 6e91b76
1 parent 7d32f7c
commit 6e91b76
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -1,12 +1,12 @@
 repos:
 - repo: https://github.com/ambv/black
-  rev: 22.1.0
+  rev: 23.3.0
   hooks:
   - id: black
-    language_version: python3.7
+    language_version: python3.9
     exclude: (.*_pb2.py|.*_pb2_grpc.py)
 - repo: https://github.com/timothycrosley/isort
-  rev: 5.10.1
+  rev: 5.12.0
   hooks:
     - id: isort
 - repo: https://github.com/pycqa/flake8

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,4 +1,4 @@
 [tool.black]
 line-length = 120
-target-version = ['py37']
+target-version = ['py39']
 exclude = '(vendor|.*_pb2.py|.*_pb2_grpc.py)'
diff --git a/tiktorch/server/device_pool.py b/tiktorch/server/device_pool.py
@@ -134,7 +134,7 @@ def list_devices(self) -> List[IDevice]:
             if torch.cuda.is_available():
                 ids += [f"cuda:{idx}" for idx in range(torch.cuda.device_count())]
             try:
-                if torch.backends.mps.is_available:
+                if torch.backends.mps.is_available():
                     ids += ["mps"]
             except AttributeError:
                 pass