Skip to content

Commit

Permalink
Drop unnecessary dependencies (#60)
Browse files Browse the repository at this point in the history
Not strictly required for #57 but related work.

`meshio` and `tensorboardX` are never actually used in the code.

`torch` is used, but it should be installed for the desired architecture
before attempting to install UM2N.
  • Loading branch information
jwallwork23 authored Nov 23, 2024
1 parent 567bee3 commit 05b98e9
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 3 deletions.
8 changes: 8 additions & 0 deletions UM2N/__init__.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
5 changes: 2 additions & 3 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -10,16 +10,15 @@ dependencies = [
"jupyter",
"ipython",
"matplotlib",
"meshio",
"numpy",
"pandas",
"pytest",
"rich",
"ruff",
"seaborn",
"tensorboardX",
"torch",
# "torch", # NOTE: Should be installed beforehand
"torch_geometric",
"wandb",
]
authors = [
{name = "Chunyang Wang"},
Expand Down

0 comments on commit 05b98e9

Please sign in to comment.