Fixed the most embarrassingly obvious bug

huggingface · Aug 30, 2021 · 4b9cfb5 · 4b9cfb5
1 parent 582a95c
commit 4b9cfb5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/transformers/data/data_collator.py b/src/transformers/data/data_collator.py
@@ -36,11 +36,11 @@ def __call__(self, features, return_tensors=None):
         if return_tensors is None:
             return_tensors = self.return_tensors
         if return_tensors == "tf":
-            return tf_default_data_collator(features)
+            return self.tf_call(features)
         elif return_tensors == "pt":
-            return torch_default_data_collator(features)
+            return self.torch_call(features)
         elif return_tensors == "np":
-            return numpy_default_data_collator(features)
+            return self.numpy_call(features)
         else:
             raise ValueError(f"Framework '{return_tensors}' not recognized!")