diff --git a/scripts/createdataset.py b/scripts/createdataset.py index dd7cfd9b40a644ba63241153ab4797976b17d451..81fc6a7365859b87e0a5ecaf9b835cd2c581f5ea 100644 --- a/scripts/createdataset.py +++ b/scripts/createdataset.py @@ -15,7 +15,7 @@ if __name__ == "__main__": filenames = [f for f in os.listdir(args.data) if os.path.isfile(os.path.join(args.data, f))] charge_filenames = [f for f in filenames if f.startswith("charge") or f.startswith("single_cell_")] - spectra_filenames = [f for f in filenames if not f.startswith("charge") and not f.startswith("single_cell_") and f != "expiramentlog.csv"] + spectra_filenames = [f for f in filenames if not f.startswith("charge") and not f.startswith("single_cell_") and not f.startswith("voltage_equlaization_") and f != "expiramentlog.csv"] print(f"found {len(spectra_filenames)} spectra") print(f"found {len(charge_filenames)} charge/discharge sequences")