better NaN handling in keplerian()

richteague · Aug 27, 2021 · 69e0478 · 69e0478
1 parent 3dd147b
commit 69e0478
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/gofish/__init__.py b/gofish/__init__.py
@@ -2,5 +2,5 @@
 
 from .gofish import imagecube
 
-__version__ = '1.4.1'
+__version__ = '1.4.1-1'
 __all__ = ["imagecube"]
diff --git a/gofish/gofish.py b/gofish/gofish.py
@@ -1700,7 +1700,7 @@ def keplerian(self, inc, PA, mstar, dist, x0=0.0, y0=0.0, vlsr=0.0,
                                                z_func=z_func,
                                                shadowed=shadowed)
         r_min = 0.0 if r_min is None else r_min
-        r_max = rvals.max() if r_max is None else r_max
+        r_max = np.nanmax(rvals) if r_max is None else r_max
         assert r_min < r_max, 'r_min >= r_max'
         r_m = rvals * dist * sc.au
         z_m = 0.0 if cylindrical else zvals * dist * sc.au

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setuptools.setup(
     name="gofish",
-    version="1.4.1",
+    version="1.4.1-1",
     author="Richard Teague",
     author_email="[email protected]",
     description="Fishing for molecular line emission in protoplanetary disks.",