gs: add {_strip,unstrip}_protocol

fixes resolution of relative paths fixes iterative/dvc#7638
iterative · Jun 7, 2022 · 1cca0ca · 1cca0ca
1 parent feecfba
commit 1cca0ca
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/dvc_objects/fs/implementations/gs.py b/src/dvc_objects/fs/implementations/gs.py
@@ -23,3 +23,12 @@ def fs(self):
         from gcsfs import GCSFileSystem
 
         return GCSFileSystem(**self.fs_args)
+
+    @classmethod
+    def _strip_protocol(cls, path: str) -> str:
+        from fsspec.utils import infer_storage_options
+
+        return infer_storage_options(path)["path"]
+
+    def unstrip_protocol(self, path: str) -> str:
+        return "gs://" + path.lstrip("/")