ctheodoris
/

Geneformer

Inference Endpoints

Model card Files Files and versions Community

ctheodoris commited on Jun 7

Commit

344f263

•

1 Parent(s): 4b4547f

reinstate save_to_disk patch

Files changed (1) hide show

geneformer/tokenizer.py +1 -1

geneformer/tokenizer.py CHANGED Viewed

@@ -175,7 +175,7 @@ class TranscriptomeTokenizer:
         )
         output_path = (Path(output_directory) / output_prefix).with_suffix(".dataset")
-        tokenized_dataset.save_to_disk(output_path)
     def tokenize_files(
         self, data_directory, file_format: Literal["loom", "h5ad"] = "loom"

         )
         output_path = (Path(output_directory) / output_prefix).with_suffix(".dataset")
+        tokenized_dataset.save_to_disk(str(output_path))
     def tokenize_files(
         self, data_directory, file_format: Literal["loom", "h5ad"] = "loom"