Christina Theodoris commited on
Commit
f17618a
·
1 Parent(s): f953554

update collator with dummy save fn for transformers compatibility

Browse files
geneformer/collator_for_classification.py CHANGED
@@ -103,6 +103,9 @@ class PrecollatorForGeneAndCellClassification(SpecialTokensMixin):
103
  def pad_token_id(self):
104
  return self._pad_token_id
105
 
 
 
 
106
  def _get_padding_truncation_strategies(
107
  self,
108
  padding=True,
 
103
  def pad_token_id(self):
104
  return self._pad_token_id
105
 
106
+ def save_pretrained(self, save_directory):
107
+ pass
108
+
109
  def _get_padding_truncation_strategies(
110
  self,
111
  padding=True,