lamin_cli 0.13.1__tar.gz → 0.14.0__tar.gz

{lamin_cli-0.13.1 → lamin_cli-0.14.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lamin_cli
-Version: 0.13.1
+Version: 0.14.0
 Summary: Lamin CLI.
 Author-email: Lamin Labs <open-source@lamin.ai>
 Description-Content-Type: text/markdown

lamin_cli-0.14.0/lamin_cli/__init__.py

@@ -0,0 +1,3 @@
+"""Lamin CLI."""
+
+__version__ = "0.14.0"

{lamin_cli-0.13.1 → lamin_cli-0.14.0}/lamin_cli/_get.py

@@ -5,10 +5,13 @@ import lamindb_setup as ln_setup
 
 
 def decompose_url(url: str) -> Tuple[str, str, str]:
-    assert "transform" in url
-    uid = url.split("transform/")[1]
+    assert "transform" in url or "artifact" in url
+    for entity in ["transform", "artifact"]:
+        if entity in url:
+            break
+    uid = url.split(f"{entity}/")[1]
     instance_slug = "/".join(url.replace("https://lamin.ai/", "").split("/")[:2])
-    return instance_slug, "transform", uid
+    return instance_slug, entity, uid
 
 
 def get(url: str):
@@ -28,12 +31,14 @@ def get(url: str):
     ln.settings.verbosity = "success"
 
     if entity == "transform":
-        transform = ln.Transform.filter(uid=uid).one()
+        transform = ln.Transform.get(uid)
         filepath_cache = transform.source_code.cache()
         target_filename = transform.key
         if not target_filename.endswith(transform.source_code.suffix):
             target_filename += transform.source_code.suffix
         filepath_cache.rename(target_filename)
         logger.success(f"cached source code of transform {uid} as {target_filename}")
-    else:
-        raise NotImplementedError
+    elif entity == "artifact":
+        artifact = ln.Artifact.get(uid)
+        cache_path = artifact.cache()
+        logger.success(f"cached artifact {artifact} here:\n{cache_path}")

{lamin_cli-0.13.1 → lamin_cli-0.14.0}/lamin_cli/_save.py

@@ -100,4 +100,5 @@ def save_from_filepath_cli(
             transform=transform,
             filepath=filepath,
             transform_family=transform_family,
+            from_cli=True,
         )

{lamin_cli-0.13.1 → lamin_cli-0.14.0}/tests/test_get.py

@@ -11,7 +11,7 @@ def test_decompose_url():
     assert uid == "1GCKs8zLtkc85zKv"
 
 
-def test_get():
+def test_get_transform():
     result = subprocess.run(
         "lamin get"
         " 'https://lamin.ai/laminlabs/arrayloader-benchmarks/transform/1GCKs8zLtkc85zKv'",  # noqa
@@ -19,3 +19,13 @@ def test_get():
         capture_output=True,
     )
     assert result.returncode == 0
+
+
+def test_get_artifact():
+    result = subprocess.run(
+        "lamin get"
+        " 'https://lamin.ai/laminlabs/lamin-site-assets/artifact/e2G7k9EVul4JbfsEYAy5'",  # noqa
+        shell=True,
+        capture_output=True,
+    )
+    assert result.returncode == 0

{lamin_cli-0.13.1 → lamin_cli-0.14.0}/tests/test_save_notebooks.py

@@ -93,7 +93,6 @@ def test_save_consecutive():
         env=env,
     )
     assert result.returncode == 0
-    assert "saved transform" in result.stdout.decode()
 
     # now, we have the associated artifacts
     transform = ln.Transform.filter(uid="hlsFXswrJjtt5zKv").one_or_none()
@@ -127,7 +126,6 @@ def test_save_consecutive():
         env=env,
     )
     assert result.returncode == 0
-    assert "saved transform" in result.stdout.decode()
     # the source code is overwritten with the edits, reflected in a new hash
     transform = ln.Transform.get("hlsFXswrJjtt5zKv")
     assert transform.latest_report.path.exists()

{lamin_cli-0.13.1 → lamin_cli-0.14.0}/tests/test_save_scripts.py

@@ -54,6 +54,20 @@ def test_run_save_cache():
     assert "loaded: Transform" in result.stdout.decode()
     assert "m5uCHTTpJnjQ5zKv" in result.stdout.decode()
     assert "loaded: Run" in result.stdout.decode()
+    assert "source code is already saved" in result.stdout.decode()
+
+    # you can re-save the script
+    result = subprocess.run(
+        f"lamin save {filepath}",
+        shell=True,
+        capture_output=True,
+        env=env,
+    )
+    print(result.stdout.decode())
+    print(result.stderr.decode())
+    assert result.returncode == 0
+    assert "source code is already saved" in result.stdout.decode()
+    assert "run.environment is already saved" in result.stdout.decode()
 
     # edit the script
     content = filepath.read_text() + "\n # edited"

lamin_cli-0.13.1/lamin_cli/__init__.py

@@ -1,3 +0,0 @@
-"""Lamin CLI."""
-
-__version__ = "0.13.1"