lamin_cli 1.4.1__py2.py3-none-any.whl → 1.4.2__py2.py3-none-any.whl

lamin_cli/__init__.py

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

lamin_cli/__main__.py

@@ -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/_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.dist-info → lamin_cli-1.4.2.dist-info}/METADATA

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

{lamin_cli-1.4.1.dist-info → lamin_cli-1.4.2.dist-info}/RECORD

@@ -1,14 +1,14 @@
-lamin_cli/__init__.py,sha256=4_vIM7YadBIkILkcPZp7oQ7MWMPbKniL-MvG9Pp3UYE,40
-lamin_cli/__main__.py,sha256=1GdJbaTQQmy_xPu-6QWglewBVkXdlGe37PbzSWPKCBM,12813
+lamin_cli/__init__.py,sha256=a2cxUd26lWX3nVDezWjl4gCzUH4M4on-qIOC0gNE59s,40
+lamin_cli/__main__.py,sha256=dlECexOeCvCm0nlW-B69wVD0bZzaJ3rWex6egfxV1i4,13357
 lamin_cli/_cache.py,sha256=oplwE8AcS_9PYptQUZxff2qTIdNFS81clGPkJNWk098,800
 lamin_cli/_load.py,sha256=ow9IPQog_mPvjoGj1Bn08l4AG1YyRNnEyBpQvQkfLZs,8182
 lamin_cli/_migration.py,sha256=xQi6mwnpBzY5wcv1-TJhveD7a3XJIlpiYx6Z3AJ1NF0,1063
-lamin_cli/_save.py,sha256=LvxEIXKgUPvpwhjgRyxi1MTDRcL6rlAMbMQLfPtfDhs,9047
+lamin_cli/_save.py,sha256=wdTMmALkkNUpleUfXkNbh2Ila_pUIi5vhVbr30wLhEA,10069
 lamin_cli/_settings.py,sha256=O2tecCf5EIZu98ima4DTJujo4KuywckOLgw8c-Ke3dY,1142
 lamin_cli/compute/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lamin_cli/compute/modal.py,sha256=QnR7GyyvWWWkLnou95HxS9xxSQfw1k-SiefM_qRVnU0,6010
-lamin_cli-1.4.1.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
-lamin_cli-1.4.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-lamin_cli-1.4.1.dist-info/WHEEL,sha256=ssQ84EZ5gH1pCOujd3iW7HClo_O_aDaClUbX4B8bjKY,100
-lamin_cli-1.4.1.dist-info/METADATA,sha256=dsWrH1faO8QjfAYKW7bNArDHztc7UL7ptaE6p5_NTaw,337
-lamin_cli-1.4.1.dist-info/RECORD,,
+lamin_cli-1.4.2.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
+lamin_cli-1.4.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+lamin_cli-1.4.2.dist-info/WHEEL,sha256=ssQ84EZ5gH1pCOujd3iW7HClo_O_aDaClUbX4B8bjKY,100
+lamin_cli-1.4.2.dist-info/METADATA,sha256=sCS5vIk9ZwaAbmQ4nShVxCTKu5udT7gIw7ASXW8Ry34,337
+lamin_cli-1.4.2.dist-info/RECORD,,