X-Git-Url: https://code.communitydata.science/cdsc_reddit.git/blobdiff_plain/7b14db67de8650e4858d3f102fbeab813a30ee29..7b130a30af863dfa727d80d9fea23648dcc9d5d8:/clustering/lsi_base.py diff --git a/clustering/lsi_base.py b/clustering/lsi_base.py index f07bca6..80b7101 100644 --- a/clustering/lsi_base.py +++ b/clustering/lsi_base.py @@ -18,10 +18,11 @@ class lsi_grid_sweep(grid_sweep): self.subsweep = subsweep inpath = Path(inpath) if lsi_dimensions == 'all': - lsi_paths = list(inpath.glob("*")) + lsi_paths = list(inpath.glob("*.feather")) else: lsi_paths = [inpath / (str(dim) + '.feather') for dim in lsi_dimensions] + print(lsi_paths) lsi_nums = [int(p.stem) for p in lsi_paths] self.hasrun = False self.subgrids = [self.subsweep(lsi_path, outpath, lsi_dim, *args, **kwargs) for lsi_dim, lsi_path in zip(lsi_nums, lsi_paths)]