lamin_cli 1.4.1__tar.gz → 1.5.0__tar.gz

{lamin_cli-1.4.1 → lamin_cli-1.5.0}/PKG-INFO

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

lamin_cli-1.5.0/lamin_cli/__init__.py

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

{lamin_cli-1.4.1 → lamin_cli-1.5.0}/lamin_cli/__main__.py

@@ -195,7 +195,7 @@ def connect(instance: str):
     from lamindb_setup import settings as settings_
 
     settings_.auto_connect = True
-    return connect_(instance, _reload_lamindb=False)
+    return connect_(instance, _reload_lamindb=False, _write_settings=True)
 
 
 @main.command()
@@ -270,7 +270,7 @@ def load(entity: str, uid: str | None = None, key: str | None = None, with_env:
         #     f"! please use: lamin connect {entity}"
         # )
         settings_.auto_connect = True
-        return connect(entity, _reload_lamindb=False)
+        return connect(entity, _reload_lamindb=False, _write_settings=True)
     else:
         from lamin_cli._load import load as load_
 
@@ -314,8 +314,10 @@ def get(entity: str, uid: str | None = None, key: str | None = None):
 @click.option("--description", type=str, default=None, help="A description of the artifact or transform.")
 @click.option("--stem-uid", type=str, default=None, help="The stem uid of the artifact or transform.")
 @click.option("--project", type=str, default=None, help="A valid project name or uid.")
+@click.option("--space", type=str, default=None, help="A valid space name or uid.")
+@click.option("--branch", type=str, default=None, help="A valid branch name or uid.")
 @click.option("--registry", type=str, default=None, help="Either 'artifact' or 'transform'. If not passed, chooses based on path suffix.")
-def save(path: str, key: str, description: str, stem_uid: str, project: str, registry: str):
+def save(path: str, key: str, description: str, stem_uid: str, project: str, space: str, branch: str, registry: str):
     """Save a file or folder.
 
     Example: Given a valid project name "my_project".
@@ -324,13 +326,16 @@ def save(path: str, key: str, description: str, stem_uid: str, project: str, reg
     lamin save my_table.csv --key my_tables/my_table.csv --project my_project
     ```
 
+    By passing a `--project` identifier, the artifact will be labeled with the corresponding project.
+    If you pass a `--space` or `--branch` identifier, you save the artifact in the corresponding {class}`~lamindb.Space` or on the corresponding {class}`~lamindb.Branch`.
+
     Note: Defaults to saving `.py`, `.ipynb`, `.R`, `.Rmd`, and `.qmd` as {class}`~lamindb.Transform` and
     other file types and folders as {class}`~lamindb.Artifact`. You can enforce saving a file as
     an {class}`~lamindb.Artifact` by passing `--registry artifact`.
     """
     from lamin_cli._save import save_from_path_cli
 
-    if save_from_path_cli(path, key, description, stem_uid, project, registry) is not None:
+    if save_from_path_cli(path=path, key=key, description=description, stem_uid=stem_uid, project=project, space=space, branch=branch, registry=registry) is not None:
         sys.exit(1)
 
 

{lamin_cli-1.4.1 → lamin_cli-1.5.0}/lamin_cli/_save.py

@@ -52,6 +52,8 @@ def save_from_path_cli(
     description: str | None,
     stem_uid: str | None,
     project: str | None,
+    space: str | None,
+    branch: str | None,
     registry: str | None,
 ) -> str | None:
     import lamindb_setup as ln_setup
@@ -97,6 +99,20 @@ def save_from_path_cli(
             raise ln.errors.InvalidArgument(
                 f"Project '{project}' not found, either create it with `ln.Project(name='...').save()` or fix typos."
             )
+    if space is not None:
+        space_record = ln.Space.filter(ln.Q(name=space) | ln.Q(uid=space)).one_or_none()
+        if space_record is None:
+            raise ln.errors.InvalidArgument(
+                f"Space '{space}' not found, either create it on LaminHub or fix typos."
+            )
+    if branch is not None:
+        branch_record = ln.Branch.filter(
+            ln.Q(name=branch) | ln.Q(uid=branch)
+        ).one_or_none()
+        if branch_record is None:
+            raise ln.errors.InvalidArgument(
+                f"Branch '{branch}' not found, either create it with `ln.Branch(name='...').save()` or fix typos."
+            )
 
     is_cloud_path = not isinstance(path, LocalPathClasses)
 
@@ -120,9 +136,12 @@ def save_from_path_cli(
             logger.error("Please pass a key or description via --key or --description")
             return "missing-key-or-description"
 
-        artifact = ln.Artifact(
-            path, key=key, description=description, revises=revises
-        ).save()
+        artifact = ln.Artifact(path, key=key, description=description, revises=revises)
+        if space is not None:
+            artifact.space = space_record
+        if branch is not None:
+            artifact.branch = branch_record
+        artifact.save()
         logger.important(f"saved: {artifact}")
         logger.important(f"storage path: {artifact.path}")
         if ln_setup.settings.storage.type == "s3":
@@ -208,7 +227,12 @@ def save_from_path_cli(
                 key=path.name,
                 type="script" if path.suffix in {".R", ".py"} else "notebook",
                 revises=revises,
-            ).save()
+            )
+            if space is not None:
+                transform.space = space_record
+            if branch is not None:
+                transform.branch = branch_record
+            transform.save()
             logger.important(f"created Transform('{transform.uid}')")
         if project is not None:
             transform.projects.add(project_record)

{lamin_cli-1.4.1 → lamin_cli-1.5.0}/tests/core/test_load.py

@@ -21,6 +21,17 @@ def test_load_transform():
     import lamindb_setup as ln_setup
 
     print(ln_setup.settings.instance.slug)
+
+    # check via a renamed instance
+    result = subprocess.run(
+        "lamin load"
+        " 'https://lamin.ai/laminlabs/lamin-dev1072025/transform/EWKgIa9dJB0n'"
+        " --with-env",
+        shell=True,
+        capture_output=True,
+    )
+    assert result.returncode == 0
+
     result = subprocess.run(
         "lamin load"
         " 'https://lamin.ai/laminlabs/lamin-dev/transform/VFYCIuaw2GsX0000'"
@@ -39,6 +50,8 @@ def test_load_transform():
     assert path1.exists()
     assert path2.exists()
 
+    subprocess.run("lamin connect laminlabs/lamin-dev", shell=True)
+
     # below will fail because it will say "these files already exist"
     result = subprocess.run(
         "lamin load transform --uid VFYCIuaw2GsX --with-env",
@@ -77,6 +90,8 @@ def test_get_load_artifact():
     )
     assert result.returncode == 0
 
+    subprocess.run("lamin connect laminlabs/lamin-site-assets", shell=True)
+
     # partial uid
     result = subprocess.run(
         "lamin load artifact --uid e2G7k9EVul4JbfsEYA",

{lamin_cli-1.4.1 → lamin_cli-1.5.0}/tests/core/test_save_files.py

@@ -24,10 +24,12 @@ def test_save_local_file():
     )
     assert result.returncode == 1
 
-    ln.Project(name="test_project").save()
+    project = ln.Project(name="test_project").save()
+    # cannot define Space with regular user, is defined in lamindb/tests/permissions
+    branch = ln.Branch(name="contrib1").save()
 
     result = subprocess.run(
-        f"lamin save {filepath} --key mytest --project test_project",
+        f"lamin save {filepath} --key mytest --project test_project --branch contrib1",
         shell=True,
         capture_output=True,
     )
@@ -38,6 +40,10 @@ def test_save_local_file():
     assert "labeled with project: test_project" in result.stdout.decode()
     assert result.returncode == 0
 
+    artifact = ln.Artifact.get(key="mytest")
+    assert artifact.branch == branch
+    assert project in artifact.projects.all()
+
     # test passing the registry and saving the same file
     result = subprocess.run(
         f"lamin save {filepath} --key mytest --registry artifact",

lamin_cli-1.4.1/lamin_cli/__init__.py

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