Ignore if torch is not installed

NKI-AI · Sep 23, 2023 · 788a77c · 788a77c
1 parent e213e90
commit 788a77c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 102 deletions.
diff --git a/dlup/utils/__init__.py b/dlup/utils/__init__.py
@@ -7,7 +7,7 @@
 from dlup.utils.imports import PYTORCH_AVAILABLE
 
 if PYTORCH_AVAILABLE:
-    import torch  # type: ignore
+    import torch  # type: ignore # pylint: disable=import-error
 
 
 class ArrayEncoder(json.JSONEncoder):

diff --git a/tests/new_test.py b/tests/new_test.py
diff --git a/tests/utils/test_utils.py b/tests/utils/test_utils.py
@@ -1,3 +1,5 @@
+# Copyright (c) dlup contributors
+
 import json
 
 import numpy as np