adding relative path code to generate_waveforms

ML4GW · Aug 29, 2022 · ff15493 · ff15493
1 parent 59a7544
commit ff15493
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/projects/sandbox/generate_waveforms/generate_waveforms.py b/projects/sandbox/generate_waveforms/generate_waveforms.py
@@ -17,7 +17,7 @@
 
 @typeo
 def main(
-    prior_file: str,
+    prior_file: Path,
     n_samples: int,
     logdir: Path,
     datadir: Path,
@@ -56,6 +56,11 @@ def main(
         )
         return signal_file
 
+    # if prior file is a relative path,
+    # make it relative to this script
+    if not prior_file.is_absolute():
+        prior_file = Path(__file__).resolve().parent / prior_file
+
     # log and print out some simulation parameters
     logging.info("Simulation parameters")
     logging.info("Number of samples     : {}".format(n_samples))

diff --git a/projects/sandbox/pyproject.toml b/projects/sandbox/pyproject.toml
@@ -122,7 +122,7 @@ verbose = true
 commands.resnet = "${base.resnet}"
 
 [tool.typeo.scripts.timeslide_injections]
-start = 1262620622 
+start = "${base.stop}" 
 stop = 1262751694
 datadir = "${base.basedir}/timeslide_injections/"
 logdir = "${base.logdir}"