Deep Learning Workshop

@@ -67,7 +67,7 @@ if __name__ == "__main__":
     with open(os.path.join(args.inputPath, args.inputFile), mode="r") as tabfile:
         df = pd.read_csv(tabfile, delimiter='\t')
         print("All rows in df: {}".format(len(df.index)))
-        df_filtered = df.loc[df['label'] in ["exon", "utr"]]
+        df_filtered = df.loc[df['label'] == "exon" or df['label'] == "utr"]
         print("Only exon and utr rows in df: {}".format(len(df_filtered.index)))
         # print("df: {}".format(df))
         sequences = df_filtered['sequence']