diff --git a/gwemopt/args.py b/gwemopt/args.py index a5f7316..6aceb9b 100644 --- a/gwemopt/args.py +++ b/gwemopt/args.py @@ -1,6 +1,7 @@ """ Module for parsing command line arguments """ + import argparse from gwemopt.paths import CATALOG_DIR, DEFAULT_LIGHTCURVE_DIR diff --git a/gwemopt/coverage.py b/gwemopt/coverage.py index 832a538..d3fa0b5 100644 --- a/gwemopt/coverage.py +++ b/gwemopt/coverage.py @@ -803,24 +803,24 @@ def timeallocation(params, map_struct, tile_structs, previous_coverage_struct=No params_hold["config"][telescope]["exposurelist"] = exposurelist if params["timeallocationType"] == "absmag": - tile_structs_hold[ - telescope - ] = gwemopt.tiles.absmag_tiles_struct( - params_hold, - config_struct, - telescope, - map_struct, - tile_structs_hold[telescope], + tile_structs_hold[telescope] = ( + gwemopt.tiles.absmag_tiles_struct( + params_hold, + config_struct, + telescope, + map_struct, + tile_structs_hold[telescope], + ) ) elif params["timeallocationType"] == "powerlaw": - tile_structs_hold[ - telescope - ] = gwemopt.tiles.powerlaw_tiles_struct( - params_hold, - config_struct, - telescope, - map_struct, - tile_structs_hold[telescope], + tile_structs_hold[telescope] = ( + gwemopt.tiles.powerlaw_tiles_struct( + params_hold, + config_struct, + telescope, + map_struct, + tile_structs_hold[telescope], + ) ) if params["timeallocationType"] == "absmag": diff --git a/gwemopt/paths.py b/gwemopt/paths.py index 8c69d3a..9857c03 100644 --- a/gwemopt/paths.py +++ b/gwemopt/paths.py @@ -1,6 +1,7 @@ """ Module for defining paths to data directories. """ + from pathlib import Path DATA_DIR = Path(__file__).parent.absolute().joinpath("data")