[snapshot] add verification before redoing pspsec

This commit is contained in:
Noe Brucy
2022-05-27 15:17:22 +02:00
parent bab0657b02
commit 6c912a5294
+3 -2
View File
@@ -1371,9 +1371,10 @@ class SnapshotProcessor(HDF5Container):
df = pd.read_csv(csv_name, header=None, names=header) df = pd.read_csv(csv_name, header=None, names=header)
return {key: df[key].values for key in df} return {key: df[key].values for key in df}
def _pspec(self, **kwargs): def _pspec(self, overwrite_file=False, **kwargs):
outfile = self.pspec_filename outfile = self.pspec_filename
pspec_new.pspec(repo=self.path, iouts=[self.num], outfile=outfile, **kwargs) if overwrite_file or not os.path.exists(self.pspec_filename):
pspec_new.pspec(repo=self.path, iouts=[self.num], outfile=outfile, **kwargs)
return np.array([self.pspec_filename]) return np.array([self.pspec_filename])
def _write_particles(self): def _write_particles(self):