Path reference. Ruff formatting.

Author: delucchi-cmu

benchmarks/benchmarks.py

@@ -1,5 +1,6 @@
 from lsdb_rubin.rsp_tests.critical_functions import critical_functions
 from lsdb_rubin.rsp_tests.random_access import random_access
+from pathlib import Path
 
 
 def time_critical_functions():
@@ -17,4 +18,5 @@ def time_random_access():
     cause the benchmark test to fail.
 
     See https://github.com/lsst-sitcom/linccf/blob/main/RSP/random_access.ipynb"""
-    random_access(None)
+
+    random_access(Path(__file__).parent.parent / "tests" / "data" / "mock_dp1_1000")

docs/notebooks/plot_light_curves.ipynb

@@ -707,7 +707,7 @@
     }
    ],
    "source": [
-    "plot_light_curve(dia_object.iloc[0][\"diaSource\"], filter_linestyles=plot_linestyles_none | {\"r\":\"--\" })"
+    "plot_light_curve(dia_object.iloc[0][\"diaSource\"], filter_linestyles=plot_linestyles_none | {\"r\": \"--\"})"
    ]
   },
   {

docs/notebooks/tract_patch_search.ipynb

@@ -50,7 +50,9 @@
     "# Load GAIA DR3 data.\n",
     "\n",
     "gaia = lsdb.read_hats(\n",
-    "    \"https://data.lsdb.io/hats/gaia_dr3/gaia\", margin_cache=\"https://data.lsdb.io/hats/gaia_dr3/gaia_10arcs\", columns=[\"ra\", \"dec\", \"source_id\"]\n",
+    "    \"https://data.lsdb.io/hats/gaia_dr3/gaia\",\n",
+    "    margin_cache=\"https://data.lsdb.io/hats/gaia_dr3/gaia_10arcs\",\n",
+    "    columns=[\"ra\", \"dec\", \"source_id\"],\n",
     ")\n",
     "gaia"
    ]
@@ -98,10 +100,7 @@
     "\n",
     "lsdb.catalog.Catalog.tract_patch_search = tract_patch_search\n",
     "\n",
-    "gaia_tract = gaia.tract_patch_search(\n",
-    "    skymap_reader=lsst_skymap,\n",
-    "    tract=tract_index\n",
-    ")\n",
+    "gaia_tract = gaia.tract_patch_search(skymap_reader=lsst_skymap, tract=tract_index)\n",
     "gaia_tract"
    ]
   },

src/lsdb_rubin/rsp_tests/critical_functions.py

@@ -47,5 +47,6 @@ def sum_id(df):
     ### Cell 6
     tmp_path.cleanup()
 
+
 if __name__ == "__main__":
-    critical_functions()
+    critical_functions()

src/lsdb_rubin/rsp_tests/random_access.py

@@ -6,7 +6,7 @@
 import sys
 
 
-def random_access(catalog_path):
+def random_access(catalog_path, verbose=False):
     """Run the contents of the `HATS Data Preview 1 on RSP` notebook.
     Any deviation from expected values will result in assertion failures.
 
@@ -26,15 +26,18 @@ def random_access(catalog_path):
     partition_indices = []
     for percentile in [10, 50, 90]:
         q = np.percentile(pixel_counts, percentile)
-        print(f"Percentile: {percentile}, Quartile: {q}")
+        if verbose:
+            print(f"Percentile: {percentile}, Quartile: {q}")
         index = int(np.argmin(np.abs(pixel_counts - q)))
         closest_value = pixel_counts.iloc[index]
-        print(f"Closest value: {closest_value}, partition index: {index}")
+        if verbose:
+            print(f"Closest value: {closest_value}, partition index: {index}")
         partition_indices.append(index)
 
     ### Cell 4
     for index in partition_indices:
-        print(f"Sampling partition {index} of size {pixel_counts.iloc[index]}")
+        if verbose:
+            print(f"Sampling partition {index} of size {pixel_counts.iloc[index]}")
         object_collection.sample(index, n=100, seed=10)