set number of spectra per subint from your_writer

thepetabyteproject · Apr 8, 2021 · 57cfadd · 57cfadd
1 parent 2504179
commit 57cfadd
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/bin/your_writer.py b/bin/your_writer.py
@@ -150,6 +150,14 @@
         type=int,
         default=1,
     )
+    parser.add_argument(
+        "-npsub",
+        "--nspectra_per_subint",
+        help="Number of spectra per subint (used only for writing psrfits)",
+        required=False,
+        type=int,
+        default=4032,
+    )
     parser.add_argument(
         "--replacement_policy",
         type=str,
@@ -222,7 +230,7 @@
     )
 
     if values.type == "fits":
-        w.to_fits()
+        w.to_fits(npsub=values.npsub)
     elif values.type == "fil":
         w.to_fil()
     else: