spec: wrap async and sync overridden adlfs methods

iterative · Jun 21, 2023 · 9e67bb4 · 9e67bb4
1 parent b5a4d29
commit 9e67bb4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/dvc_azure/spec.py b/dvc_azure/spec.py
@@ -8,6 +8,14 @@ def put_file(self, *args, **kwargs) -> None:
         kwargs["overwrite"] = True
         return super().put_file(*args, **kwargs)
 
+    async def _put_file(self, *args, **kwargs) -> None:
+        kwargs["overwrite"] = True
+        return await super()._put_file(*args, **kwargs)
+
     def rm(self, *args, **kwargs) -> None:
         kwargs["expand_path"] = False
         return super().rm(*args, **kwargs)
+
+    async def _rm(self, *args, **kwargs) -> None:
+        kwargs["expand_path"] = False
+        return await super()._rm(*args, **kwargs)