repeated cast

official/vision/image_classification/dataset_factory.py

@@ -477,7 +477,7 @@ class DatasetBuilder:
     parsed = tf.io.parse_single_example(record, keys_to_features)
 
     label = tf.reshape(parsed['image/class/label'], shape=[1])
-    label = tf.cast(label, dtype=tf.int32)
+    label = tf.cast(label, tf.int32)
 
     # Subtract one so that labels are in [0, 1000)
     label -= 1
@@ -507,7 +507,6 @@ class DatasetBuilder:
           standardize=self.config.standardize,
           dtype=self.dtype)
 
-    label = tf.cast(label, tf.int32)
     if self.config.one_hot:
       label = tf.one_hot(label, self.num_classes)
       label = tf.reshape(label, [self.num_classes])