Merge pull request #172 from lincc-frameworks/benchmark_passbands

Use Path in benchmarks
lincc-frameworks · Oct 16, 2024 · e9cd611 · e9cd611
2 parents 97d9ca6 + a018bdd
commit e9cd611
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/benchmarks/benchmarks.py b/benchmarks/benchmarks.py
@@ -5,8 +5,6 @@
 For more information on writing benchmarks:
 https://asv.readthedocs.io/en/stable/writing_benchmarks.html."""
 
-import os
-
 import numpy as np
 import tdastro
 from astropy import units as u
@@ -20,16 +18,16 @@
 
 def _load_test_passbands():
     """Load passbands to use in various benchmarks."""
-    passbands_dir = os.path.join(tdastro._TDASTRO_TEST_DATA_DIR, "passbands")
+    passbands_dir = tdastro._TDASTRO_TEST_DATA_DIR / "passbands"
     passbands = PassbandGroup(
         passband_parameters=[
             {
                 "filter_name": "g",
-                "table_path": f"{passbands_dir}/LSST/g.dat",
+                "table_path": passbands_dir / "LSST" / "g.dat",
             },
             {
                 "filter_name": "r",
-                "table_path": f"{passbands_dir}/LSST/r.dat",
+                "table_path": passbands_dir / "LSST" / "r.dat",
             },
         ],
         survey="LSST",