Add torch import check - thanks @ddundo; add missing dependency

mesh-adaptation · Nov 23, 2024 · 53694d6 · 53694d6
1 parent d5da0e6
commit 53694d6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/UM2N/__init__.py b/UM2N/__init__.py
@@ -1,5 +1,13 @@
 import os
 
+try:
+    import torch  # noqa
+except ImportError:
+    raise ImportError(
+        "PyTorch is required to use this package. Please install it first."
+        " Visit https://pytorch.org/get-started/locally/ for installation instructions."
+    )
+
 os.environ["OMP_NUM_THREADS"] = "1"
 
 from pkg_resources import DistributionNotFound, get_distribution  # noqa

diff --git a/pyproject.toml b/pyproject.toml
@@ -18,6 +18,7 @@ dependencies = [
   "seaborn",
   # "torch", # NOTE: Should be installed beforehand
   "torch_geometric",
+  "wandb",
 ]
 authors = [
   {name = "Chunyang Wang"},