Remove .tar.gz extension for the untared filename
This is not working after Keras changes.
PiperOrigin-RevId: 380909699
diff --git a/integrations/tensorflow/e2e/mobile_bert_squad_test.py b/integrations/tensorflow/e2e/mobile_bert_squad_test.py
index bc11a57..99abbbb 100644
--- a/integrations/tensorflow/e2e/mobile_bert_squad_test.py
+++ b/integrations/tensorflow/e2e/mobile_bert_squad_test.py
@@ -26,10 +26,10 @@
MAX_SEQ_LENGTH = 384 # Max input sequence length used in mobilebert_squad.
-FILE_NAME = 'mobilebert_squad_savedmodels.tar.gz'
+FILE_NAME = 'mobilebert_squad_savedmodels'
MODEL_URL = posixpath.join(
- 'https://storage.googleapis.com/cloud-tpu-checkpoints/mobilebert/',
- FILE_NAME)
+ f'https://storage.googleapis.com/cloud-tpu-checkpoints/mobilebert/{FILE_NAME}.tar.gz'
+)
class MobileBertSquadTest(tf_test_utils.TracedModuleTestCase):
