From 98773f3355bfd7eda6efe21c2df7a65970dd7442 Mon Sep 17 00:00:00 2001 From: Alejandro Velez Date: Thu, 5 Sep 2024 10:18:14 -0400 Subject: [PATCH] mend --- tdc/resource/pinnacle.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/tdc/resource/pinnacle.py b/tdc/resource/pinnacle.py index b62aae46..f5034d05 100644 --- a/tdc/resource/pinnacle.py +++ b/tdc/resource/pinnacle.py @@ -85,10 +85,14 @@ def get_exp_data(self, seed=1, split="train"): for file in file_list: os.remove(os.path.join("./data", file)) print("downloading pinancle zip data...") - zip_data_download_wrapper(filename, "./data", ["pinnacle_output{}".format(x) for x in range(1,11)]) + zip_data_download_wrapper( + filename, "./data", + ["pinnacle_output{}".format(x) for x in range(1, 11)]) print("success!") # Get non-csv files and remove them - non_csv_files = [f for f in os.listdir("./data") if not f.endswith(".csv")] + non_csv_files = [ + f for f in os.listdir("./data") if not f.endswith(".csv") + ] for x in non_csv_files: os.remove("./data/{}".format(x)) # Get a list of all CSV files in the unzipped folder @@ -111,7 +115,4 @@ def get_exp_data(self, seed=1, split="train"): df["disease"] = disease x.append(df) os.remove("./data/{}".format(file)) - return pd.concat(x, axis=0, ignore_index=True) - - - + return pd.concat(x, axis=0, ignore_index=True) \ No newline at end of file