Merge pull request #8 from hammerlab/pickle-py3

Python3 pickle error with bytes

scripts/create-combined-class1-dataset.py

@@ -68,7 +68,7 @@ if __name__ == "__main__":
     args = parser.parse_args()
 
     print("Reading %s..." % args.iedb_pickle_path)
-    with open(args.iedb_pickle_path, "r") as f:
+    with open(args.iedb_pickle_path, "rb") as f:
         iedb_datasets = pickle.load(f)
 
     print("Reading %s..." % args.netmhcpan_csv_path)

scripts/create-iedb-class1-dataset.py

@@ -84,5 +84,5 @@ if __name__ == "__main__":
                 "percent_positive",
                 "count"])
         print("# distinct pMHC entries: %d" % len(columns["mhc"]))
-    with open(OUTPUT_PATH, "w") as f:
+    with open(OUTPUT_PATH, "wb") as f:
         pickle.dump(assay_dataframes, f, pickle.HIGHEST_PROTOCOL)