Implement file uploading for Storage Client (#758)

iterative · Jan 6, 2025 · b4239e1 · b4239e1
1 parent 81c774f
commit b4239e1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/datachain/client/fsspec.py b/src/datachain/client/fsspec.py
@@ -390,6 +390,12 @@ def open_object(
             self.fs.open(self.get_full_path(file.path, file.version)), cb
         )  # type: ignore[return-value]
 
+    def upload(self, path: str, data: bytes) -> "File":
+        full_path = self.get_full_path(path)
+        self.fs.pipe_file(full_path, data)
+        file_info = self.fs.info(full_path)
+        return self.info_to_file(file_info, path)
+
     def download(self, file: "File", *, callback: Callback = DEFAULT_CALLBACK) -> None:
         sync(get_loop(), functools.partial(self._download, file, callback=callback))