datachain 0.9.0__tar.gz → 0.9.1__tar.gz

{datachain-0.9.0 → datachain-0.9.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: datachain
-Version: 0.9.0
+Version: 0.9.1
 Summary: Wrangle unstructured AI data at scale
 Author-email: Dmitry Petrov <support@dvc.org>
 License: Apache-2.0

{datachain-0.9.0 → datachain-0.9.1}/src/datachain/lib/file.py

@@ -214,10 +214,13 @@ class File(DataModel):
 
             catalog = get_catalog()
 
-        parent, name = posixpath.split(path)
+        from datachain.client.fsspec import Client
 
-        client = catalog.get_client(parent)
-        file = client.upload(data, name)
+        client_cls = Client.get_implementation(path)
+        source, rel_path = client_cls.split_url(path)
+
+        client = catalog.get_client(client_cls.get_uri(source))
+        file = client.upload(data, rel_path)
         if not isinstance(file, cls):
             file = cls(**file.model_dump())
         file._set_stream(catalog)

{datachain-0.9.0 → datachain-0.9.1}/src/datachain.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: datachain
-Version: 0.9.0
+Version: 0.9.1
 Summary: Wrangle unstructured AI data at scale
 Author-email: Dmitry Petrov <support@dvc.org>
 License: Apache-2.0

{datachain-0.9.0 → datachain-0.9.1}/tests/func/test_file.py

@@ -50,16 +50,18 @@ def test_upload(cloud_test_catalog):
 
     src_uri = ctc.src_uri
     filename = "image_1.jpg"
-    source = f"{src_uri}/upload-test-images"
+    dest = f"{src_uri}/upload-test-images"
     catalog = ctc.catalog
 
     img_bytes = b"bytes"
 
-    f = File.upload(img_bytes, f"{source}/{filename}", catalog)
+    f = File.upload(img_bytes, f"{dest}/{filename}", catalog)
 
-    assert f.path == filename
-    assert f.source == source
+    client = catalog.get_client(src_uri)
+    source, rel_path = client.split_url(f"{dest}/{filename}")
+
+    assert f.path == rel_path
+    assert f.source == client.get_uri(source)
     assert f.read() == img_bytes
 
-    client = catalog.get_client(src_uri)
-    client.fs.rm(source, recursive=True)
+    client.fs.rm(dest, recursive=True)