Merge pull request #1695 from pared/1640

yaml loading error: display relative path
iterative · Mar 13, 2019 · 64e5b1e · 64e5b1e
2 parents 5530125 + 161d3ce
commit 64e5b1e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/dvc/exceptions.py b/dvc/exceptions.py
@@ -212,6 +212,7 @@ def __init__(self):
 
 class StageFileCorruptedError(DvcException):
     def __init__(self, path):
+        path = os.path.relpath(path)
         super(StageFileCorruptedError, self).__init__(
             "unable to read stage file: {} "
             "YAML file structure is corrupted".format(path)

diff --git a/tests/test_add.py b/tests/test_add.py
@@ -415,7 +415,7 @@ def test(self):
             ret = main(["add", self.FOO])
             self.assertEqual(0, ret)
 
-            foo_stage = os.path.abspath(self.FOO + Stage.STAGE_FILE_SUFFIX)
+            foo_stage = os.path.relpath(self.FOO + Stage.STAGE_FILE_SUFFIX)
 
             # corrupt stage file
             with open(foo_stage, "a+") as file: