pe.py: make globbing of pulsar file directory recursive to be consist…

…ent with heterodyning
cwinpy · Mar 18, 2024 · 1d1d21f · 1d1d21f
1 parent 4bf53e2
commit 1d1d21f
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/cwinpy/pe/pe.py b/cwinpy/pe/pe.py
@@ -13,6 +13,7 @@
 import sys
 import warnings
 from argparse import ArgumentParser
+from pathlib import Path
 
 import bilby
 import numpy as np
@@ -1545,16 +1546,14 @@ def create_dag(self, config, **kwargs):
                 pulsars = []
                 for parfile in parfiles:
                     # add "*.par" wildcard to any directories
-                    if os.path.isdir(parfile):
-                        parfile = os.path.join(parfile, "*.par")
+                    if Path(parfile).is_dir():
+                        parfile = Path(parfile).rglob("*.par")
+                    else:
+                        parfile = [Path(parfile)]
 
                     # get all parameter files
                     pulsars.extend(
-                        [
-                            par
-                            for par in glob.glob(parfile)
-                            if os.path.splitext(par)[1] == ".par"
-                        ]
+                        [str(par) for par in parfile if par.suffix == ".par"]
                     )
 
                 # get names of all the pulsars