Merge branch 'main' into issue/import/453

Author: delucchi-cmu

docs/conf.py

@@ -24,7 +24,12 @@
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
 
-extensions = ["sphinx.ext.mathjax", "sphinx.ext.napoleon", "sphinx.ext.viewcode"]
+extensions = [
+    "sphinx.ext.mathjax",
+    "sphinx.ext.napoleon",
+    "sphinx.ext.viewcode",
+    "sphinx_design",
+]
 
 extensions.append("autoapi.extension")
 extensions.append("nbsphinx")

docs/getting_started.rst

@@ -33,7 +33,7 @@ The latest release version of HATS is available to install with `pip <https://py
                 pyenv virtualenv 3.11 hats_env
                 pyenv local hats_env
 
-    We recommend Python versions **>=3.9, <=3.12**.
+    We recommend Python versions **>=3.10, <=3.12**.
 
 HATS can also be installed from source on `GitHub <https://github.com/astronomy-commons/hats>`_.
 

docs/index.rst

@@ -44,7 +44,8 @@ Getting Started
 For the most part, we recommend accessing and processing HATS data using the `LSDB package
 <https://github.com/astronomy-commons/lsdb>`_ framework. LSDB provides a variety of utility
 functions as well as a lazy, distributed execution framework using Dask. However if you are are
-interested in using just the HATS package, you can find installation instructions at the :ref:`getting started page<getting_started>`
+interested in using just the HATS package, you can find installation instructions at the 
+:doc:`getting started page</getting_started>`.
 
 Acknowledgements
 -------------------------------------------------------------------------------

docs/requirements.txt

@@ -8,3 +8,4 @@ sphinx
 sphinx-autoapi
 sphinx-copybutton
 sphinx-book-theme
+sphinx-design

src/hats/io/parquet_metadata.py

@@ -276,5 +276,5 @@ def aggregate_column_statistics(
             "max_value": stats_lists[1],
             "null_count": stats_lists[2],
         }
-    )
+    ).set_index("column_names")
     return frame

src/hats/pixel_math/healpix_shim.py

@@ -62,8 +62,8 @@ def radec2pix(order: int, ra: float, dec: float) -> np.ndarray[np.int64]:
     if not is_order_valid(order):
         raise ValueError("Invalid value for order")
 
-    ra = Longitude(ra, unit="deg")
-    dec = Latitude(dec, unit="deg")
+    ra = Longitude(np.asarray(ra, dtype=np.float64), unit="deg")
+    dec = Latitude(np.asarray(dec, dtype=np.float64), unit="deg")
 
     return cdshealpix.lonlat_to_healpix(ra, dec, order).astype(np.int64)
 

tests/hats/pixel_math/test_healpix_shim.py

@@ -166,6 +166,22 @@ def test_radec2pix_lonlat():
         assert np.all(pixels == expected_pixels)
 
 
+def test_radec2pix_lonlat_float32():
+    orders = [0, 1, 5, 10, 20, 29]
+    ras = np.arange(-180.0, 180.0, 10.0)
+    ras_f = ras.astype(np.float32)
+    decs = np.arange(-90.0, 90.0, 180 // len(ras))
+    decs_f = decs.astype(np.float32)
+    for order in orders:
+        expected_pixels = cdshealpix.lonlat_to_healpix(
+            Longitude(ras, unit="deg"), Latitude(decs, unit="deg"), order
+        )
+        # Verify that healpixshim can work with float32 versions
+        # Fixes https://github.com/astronomy-commons/hats-import/issues/458
+        pixels = hps.radec2pix(order, ras_f, decs_f)
+        assert np.all(pixels == expected_pixels)
+
+
 def test_radec2pix_invalid():
     orders = [0, 1, 5, 10, 20, 29]
     invalid_orders = [-1000, -1, 30, 40]