pandas 1.3 fix

scverse · Jul 13, 2021 · ff67d07 · ff67d07
1 parent 334c038
commit ff67d07
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/scvi/data/_built_in_data/_pbmc.py b/scvi/data/_built_in_data/_pbmc.py
@@ -1,5 +1,4 @@
 import os
-import pickle
 from typing import List
 
 import anndata
@@ -65,9 +64,7 @@ def _load_pbmc_dataset(
         _download(urls[i], save_path, save_fns[i])
 
     de_metadata = pd.read_csv(os.path.join(save_path, "gene_info_pbmc.csv"), sep=",")
-    pbmc_metadata = pickle.load(
-        open(os.path.join(save_path, "pbmc_metadata.pickle"), "rb")
-    )
+    pbmc_metadata = pd.read_pickle(os.path.join(save_path, "pbmc_metadata.pickle"))
     pbmc8k = _load_dataset_10x(
         "pbmc8k",
         save_path=save_path,