Christina Theodoris commited on
Commit
f45a6c7
·
1 Parent(s): be98543

default id_class_dict to None for gene classifiers

Browse files
Files changed (1) hide show
  1. geneformer/classifier_utils.py +1 -1
geneformer/classifier_utils.py CHANGED
@@ -94,7 +94,7 @@ def remove_rare(data, rare_threshold, label, nproc):
94
  return data
95
 
96
 
97
- def label_classes(classifier, data, gene_class_dict, nproc, id_class_dict):
98
  if classifier == "cell":
99
  label_set = set(data["label"])
100
  elif classifier == "gene":
 
94
  return data
95
 
96
 
97
+ def label_classes(classifier, data, gene_class_dict, nproc, id_class_dict=None):
98
  if classifier == "cell":
99
  label_set = set(data["label"])
100
  elif classifier == "gene":