make pyvista optional

Resolves #5
NVlabs · Aug 21, 2024 · e7c7a88 · e7c7a88
1 parent 2e7af1b
commit e7c7a88
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,7 +26,6 @@ dependencies = [
     "einops>=0.7.0",
     "netCDF4>=1.6.5",
     "numpy>=1.23.3",
-    "pyvista>=0.43.2",
     "torch>=2.0.1",
 ]
 
@@ -35,6 +34,9 @@ dependencies = [
 
 
 [project.optional-dependencies]
+viz = [
+    "pyvista>=0.43.2",
+]
 test = [
     "pytest>=6.2.4",
     "black>=21.5b2",