Rename a class to reflect framework pattern AutoModelXxx -> TFAutoModelXxx (huggingface#16993)

amyeroberts · elusenji · commit e4be09268b00 · 2022-06-12T11:14:27.000+02:00
diff --git a/src/transformers/models/auto/modeling_tf_auto.py b/src/transformers/models/auto/modeling_tf_auto.py
@@ -411,7 +411,9 @@ class TFAutoModelForImageClassification(_BaseAutoModelClass):
     _model_mapping = TF_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING
 
 
-AutoModelForImageClassification = auto_class_update(TFAutoModelForImageClassification, head_doc="image classification")
+TFAutoModelForImageClassification = auto_class_update(
+    TFAutoModelForImageClassification, head_doc="image classification"
+)
 
 
 class TFAutoModelForVision2Seq(_BaseAutoModelClass):
