Merge pull request #7 from radionets-project/update_dependencies

Make pyvisgen dependency optional

Author: tgross03

pyproject.toml

@@ -44,7 +44,12 @@ dependencies = [
   "torch",
   "tqdm",
   "casatools",
-  "pyvisgen@git+https://github.com/radionets-project/pyvisgen.git@main",
+]
+
+[project.optional-dependencies]
+
+pyvisgen = [
+  "pyvisgen"
 ]
 
 [dependency-groups]

pyvisgrid/core/gridder.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from dataclasses import dataclass
 from pathlib import Path
 
@@ -7,7 +9,11 @@
 from astropy.io import fits
 from casatools.table import table
 from numpy.exceptions import AxisError
-from pyvisgen.simulation import Observation, Visibilities
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from pyvisgen.simulation import Observation, Visibilities
 
 import pyvisgrid.plotting as plotting
 from pyvisgrid.core.stokes import get_stokes_from_vis_data

pyvisgrid/core/stokes.py

@@ -1,5 +1,11 @@
+from __future__ import annotations
+
 import numpy as np
-from pyvisgen.simulation import Visibilities
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from pyvisgen.simulation import Visibilities
 
 
 def get_stokes_from_vis_data(