Formatting script structure

py/picca/bin/picca_Pk1D_average_mean.py

@@ -1,8 +1,11 @@
-#!/usr/bin/env python
-"""Compute the averaged 1D power spectrum
-"""
+#!/usr/bin/env python3
+"""Compute the averaged 1D power spectrum"""
+
+import argparse
+import ast
+import glob
+import sys
 
-import sys, glob, argparse, ast
 from picca.pk1d import postproc_pk1d
 from picca.utils import userprint
 
@@ -69,3 +72,8 @@ def main(cmdargs=None):
         args.output_path,
         weighted_mean=False,
     )
+
+
+if __name__ == "__main__":
+    cmdargs = sys.argv[1:]
+    main(cmdargs)

py/picca/bin/picca_Pk1D_cross_exposure.py

@@ -1,24 +1,28 @@
-#!/usr/bin/env python
-"""Compute the individual cross-exposure 1D power spectra
-"""
+#!/usr/bin/env python3
+"""Compute the individual cross-exposure 1D power spectra"""
+
+import argparse
+import glob
+import multiprocessing as mp
+import os
+import sys
+from functools import partial
 
-import os, argparse, glob
 import fitsio
 import numpy as np
-from picca.pk1d.compute_pk1d import compute_pk_cross_exposure, Pk1D
+
+from picca.pk1d.compute_pk1d import Pk1D, compute_pk_cross_exposure
 from picca.utils import userprint
-import multiprocessing as mp
-from functools import partial
 
 
 def treat_pk_file(out_dir, filename):
     """
     Takes a single file containing the FFT of delta for
     multiple exposures and computes the cross-exposure power spectrum. The function
     returns nothing, but writes to disk a new fits file with all the information needed
-    to compute Pk_cross_exposure. This is done by looping over each targetid and chunkid, 
+    to compute Pk_cross_exposure. This is done by looping over each targetid and chunkid,
     and computing Pk_cross_exposure for each pair of exposures.
-    
+
     Arguments
     ---------
     out_dir: string
@@ -196,3 +200,8 @@ def main(cmdargs=None):
     else:
         with mp.Pool(args.num_processors) as pool:
             pool.map(func, files)
+
+
+if __name__ == "__main__":
+    cmdargs = sys.argv[1:]
+    main(cmdargs)