datafs: info: raise better FileNotFoundError (#407)

iterative · Jul 27, 2023 · 4617ead · 4617ead
1 parent 1483f63
commit 4617ead
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/dvc_data/fs.py b/src/dvc_data/fs.py
@@ -115,7 +115,16 @@ def ls(self, path, detail=True, **kwargs):
 
     def info(self, path, **kwargs):
         key = self._get_key(path)
-        info = self.index.info(key)
+
+        try:
+            info = self.index.info(key)
+        except KeyError as exc:
+            raise FileNotFoundError(
+                errno.ENOENT,
+                os.strerror(errno.ENOENT),
+                path,
+            ) from exc
+
         info["name"] = path
         return info
 

diff --git a/src/dvc_data/index/index.py b/src/dvc_data/index/index.py
@@ -462,8 +462,6 @@ def info(self, key: DataIndexKey):
             entry = self[key]
         except ShortKeyError:
             entry = None
-        except KeyError as exc:
-            raise FileNotFoundError from exc
 
         return self._info_from_entry(key, entry)