Merge of PR #1698

PiperOrigin-RevId: 268729413

Author: Vooblin

tensor2tensor/utils/checkpoint_compatibility_test.py

@@ -41,8 +41,7 @@
 
 
 def get_data_dir():
-  pkg = os.path.abspath(__file__)
-  pkg, _ = os.path.split(pkg)
+  pkg, _ = os.path.split(__file__)
   pkg, _ = os.path.split(pkg)
   return os.path.join(pkg, "test_data")
 

tensor2tensor/utils/hparams_lib_test.py

@@ -30,8 +30,7 @@ class HparamsLibTest(tf.test.TestCase):
 
   def testCreateHparamsFromJson(self):
     # Get json_path
-    pkg = os.path.abspath(__file__)
-    pkg, _ = os.path.split(pkg)
+    pkg, _ = os.path.split(__file__)
     pkg, _ = os.path.split(pkg)
     json_path = os.path.join(
         pkg, "test_data", "transformer_test_ckpt", "hparams.json")

tensor2tensor/utils/trainer_lib_test.py

@@ -147,8 +147,7 @@ def body(args, mb=model.body):
 
   def testCreateHparams(self):
     # Get json_path
-    pkg = os.path.abspath(__file__)
-    pkg, _ = os.path.split(pkg)
+    pkg, _ = os.path.split(__file__)
     pkg, _ = os.path.split(pkg)
     json_path = os.path.join(
         pkg, "test_data", "transformer_test_ckpt", "hparams.json")