mend

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)