Update the V1 Eager checkpointer path so it's compatible with the Graph mode trainer

PiperOrigin-RevId: 476806931

Author: lingvo-bot

lingvo/core/checkpointer.py

@@ -541,8 +541,7 @@ def __init__(self,
     super().__init__(train_dir, models, train_params, save_only,
                      check_loading_status)
     tf.logging.info('Starting eager checkpointer v1')
-    # Distinct from EagerCheckpointerV2
-    self._train_dir = os.path.join(self._train_dir, 'ckpt_V1')
+    self._train_dir = self._train_dir
     if not tf.io.gfile.exists(self._train_dir):
       tf.io.gfile.makedirs(self._train_dir)
 

lingvo/core/checkpointer_eager_test.py

@@ -177,7 +177,7 @@ def testEagerMultiLearnerCheckpointCompatibility(self):
     checkpointer.EagerCheckpointerV1(eager_v1_logdir, mdl).Save(gsteps=0)
     checkpointer.EagerCheckpointerV2(eager_v2_logdir, mdl).Save(gsteps=0)
     eager_v1_keys = _GetCheckpointKeys(
-        os.path.join(eager_v1_logdir, 'ckpt_V1', 'ckpt-00000000'))
+        os.path.join(eager_v1_logdir, 'ckpt-00000000'))
     eager_v2_keys = _GetCheckpointKeys(
         os.path.join(eager_v2_logdir, 'ckpt_V2', 'ckpt-0'))
     # Expecting two more variables in V2 checkpoints:

lingvo/core/checkpointer_test.py

@@ -254,10 +254,7 @@ def testSaveOnly(self):
 
   def testSortCheckpointPaths(self):
     ckpts_no_pad = ['/ckpt_V2/ckpt-100', '/ckpt_V2/ckpt-20', '/ckpt_V2/ckpt-3']
-    ckpts_pad = [
-        '/ckpt_V1/ckpt-00000100', '/ckpt_V1/ckpt-00000020',
-        '/ckpt_V1/ckpt-00000003'
-    ]
+    ckpts_pad = ['/ckpt-00000100', '/ckpt-00000020', '/ckpt-00000003']
     ckpts_no_pad_sorted = checkpointer.SortCheckpointPaths(ckpts_no_pad)
     self.assertIn('3', ckpts_no_pad_sorted[0])
     self.assertIn('100', ckpts_no_pad_sorted[2])