lamin_cli 0.16.1__py2.py3-none-any.whl → 0.16.3__py2.py3-none-any.whl

lamin_cli/__init__.py

@@ -1,3 +1,3 @@
 """Lamin CLI."""
 
-__version__ = "0.16.1"
+__version__ = "0.16.3"

lamin_cli/__main__.py

@@ -52,7 +52,7 @@ else:
             },
             {
                 "name": "Configuration commands",
-                "commands": ["register", "cache", "set"],
+                "commands": ["cache", "set"],
             },
             {
                 "name": "Schema commands",
@@ -72,6 +72,7 @@ else:
         @wraps(f)
         def wrapper(*args, **kwargs):
             return f(*args, **kwargs)
+
         return wrapper
 
 
@@ -97,15 +98,22 @@ def main():
 @main.command()
 @click.argument("user", type=str)
 @click.option("--key", type=str, default=None, help="API key")
-@click.option("--password", type=str, default=None, help="legacy password")
-def login(user: str, key: Optional[str], password: Optional[str]):
-    """Login using a user email address or handle.
+def login(user: str, key: Optional[str]):
+    """Log into LaminHub.
+
+    Upon logging in the first time, you need to pass your API key via
+
+    ```
+    lamin login myemail@acme.com --key YOUR_API_KEY
+    ```
+
+    You'll find your API key in the top right corner under "Settings".
 
-    Examples: `lamin login marge` or `lamin login marge@acme.com`
+    After this, you can either use `lamin login myhandle` or `lamin login myemail@acme.com`
     """
     from lamindb_setup._setup_user import login
 
-    return login(user, key=key, password=password)
+    return login(user, key=key)
 
 
 # fmt: off
@@ -202,14 +210,6 @@ def save(filepath: str, key: str, description: str):
         sys.exit(1)
 
 
-@main.command()
-def register():
-    """Register an instance on the hub."""
-    from lamindb_setup._register_instance import register as register_
-
-    return register_()
-
-
 main.add_command(cache)
 
 

lamin_cli/_cache.py

@@ -37,4 +37,4 @@ def get_cache():
     """Get the cache directory."""
     from lamindb_setup._cache import get_cache_dir
 
-    click.echo(f"The cache directory is {get_cache_dir()}.")
+    click.echo(f"The cache directory is {get_cache_dir()}")

lamin_cli/_save.py

@@ -6,17 +6,21 @@ from lamin_utils import logger
 import re
 
 
-def get_stem_uid_and_version_from_file(file_path: Path) -> tuple[str, str]:
+def get_stem_uid_and_version_from_file(
+    file_path: Path,
+) -> tuple[str | None, str | None, str | None]:
     # line-by-line matching might be faster, but let's go with this for now
     with open(file_path) as file:
         content = file.read()
 
     if file_path.suffix == ".py":
+        uid_pattern = re.compile(r'\.context\.uid\s*=\s*["\']([^"\']+)["\']')
         stem_uid_pattern = re.compile(
             r'\.transform\.stem_uid\s*=\s*["\']([^"\']+)["\']'
         )
         version_pattern = re.compile(r'\.transform\.version\s*=\s*["\']([^"\']+)["\']')
     elif file_path.suffix == ".ipynb":
+        uid_pattern = re.compile(r'\.context\.uid\s*=\s*\\["\']([^"\']+)\\["\']')
         stem_uid_pattern = re.compile(
             r'\.transform\.stem_uid\s*=\s*\\["\']([^"\']+)\\["\']'
         )
@@ -27,20 +31,22 @@ def get_stem_uid_and_version_from_file(file_path: Path) -> tuple[str, str]:
         raise ValueError("Only .py and .ipynb files are supported.")
 
     # Search for matches in the entire file content
+    uid_match = uid_pattern.search(content)
     stem_uid_match = stem_uid_pattern.search(content)
     version_match = version_pattern.search(content)
 
     # Extract values if matches are found
+    uid = uid_match.group(1) if uid_match else None
     stem_uid = stem_uid_match.group(1) if stem_uid_match else None
     version = version_match.group(1) if version_match else None
 
-    if stem_uid is None or version is None:
+    if uid is None and (stem_uid is None or version is None):
         raise SystemExit(
-            "ln.settings.transform.stem_uid and ln.settings.transform.version aren't"
-            f" set in {file_path}\nCall ln.track() and copy/paste the output into the"
-            " notebook"
+            "ln.context.uid isn't"
+            f" set in {file_path}\nCall ln.context.track() and copy/paste the output"
+            " into the notebook"
         )
-    return stem_uid, version
+    return uid, stem_uid, version
 
 
 def save_from_filepath_cli(
@@ -55,7 +61,7 @@ def save_from_filepath_cli(
     ln_setup.settings.auto_connect = True
 
     import lamindb as ln
-    from lamindb._finish import save_run_context_core
+    from lamindb._finish import save_context_core
 
     ln_setup.settings.auto_connect = auto_connect_state
 
@@ -75,17 +81,22 @@ def save_from_filepath_cli(
         return None
     else:
         # consider notebooks & scripts a transform
-        stem_uid, transform_version = get_stem_uid_and_version_from_file(filepath)
-        # the corresponding transform family in the transform table
-        transform_family = ln.Transform.filter(uid__startswith=stem_uid).all()
-        if len(transform_family) == 0:
-            logger.error(
-                f"Did not find stem uid '{stem_uid}'"
-                " in Transform registry. Did you run ln.track()?"
-            )
-            return "not-tracked-in-transform-registry"
-        # the specific version
-        transform = transform_family.filter(version=transform_version).one()
+        uid, stem_uid, transform_version = get_stem_uid_and_version_from_file(filepath)
+        if uid is not None:
+            transform = ln.Transform.filter(uid=uid).one_or_none()
+            if transform is None:
+                logger.error(
+                    f"Did not find uid '{uid}'"
+                    " in Transform registry. Did you run ln.context.track()?"
+                )
+                return "not-tracked-in-transform-registry"
+            # refactor this, save_context_core should not depend on transform_family
+            transform_family = transform.versions
+        else:
+            # the corresponding transform family in the transform table
+            transform_family = ln.Transform.filter(uid__startswith=stem_uid).all()
+            # the specific version
+            transform = transform_family.get(version=transform_version)
         # latest run of this transform by user
         run = ln.Run.filter(transform=transform).order_by("-started_at").first()
         if run.created_by.id != ln_setup.settings.user.id:
@@ -95,7 +106,7 @@ def save_from_filepath_cli(
             )
             if response != "y":
                 return "aborted-save-notebook-created-by-different-user"
-        return save_run_context_core(
+        return save_context_core(
             run=run,
             transform=transform,
             filepath=filepath,

{lamin_cli-0.16.1.dist-info → lamin_cli-0.16.3.dist-info}/METADATA

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

lamin_cli-0.16.3.dist-info/RECORD

@@ -0,0 +1,10 @@
+lamin_cli/__init__.py,sha256=-VEBAkJGvngQDNsXIQb5tfZImdAMjJW8VcZ1T7rx834,41
+lamin_cli/__main__.py,sha256=7mUJe-6Vq8WNNR90t2j0Pa_XLMNaS9rdctj7OEfhum0,7898
+lamin_cli/_cache.py,sha256=6LrrWNrdZzH_BrohQ2n24xLFhUQ5EFan1bnzqMX_mIg,812
+lamin_cli/_get.py,sha256=MXcR9PTW0FzY3uIPa-50SLCrtuRElxgDB3Vkfo-TFT0,1616
+lamin_cli/_migration.py,sha256=dULOvbpJ4VBiXuxPAM8jFGnBkh7pQGqE5eP-UC6uxWc,1055
+lamin_cli/_save.py,sha256=H_QQzWE3fId0GtM4LiIdNcgi81Wafvjh_V9B30SweBk,4821
+lamin_cli-0.16.3.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
+lamin_cli-0.16.3.dist-info/WHEEL,sha256=Sgu64hAMa6g5FdzHxXv9Xdse9yxpGGMeagVtPMWpJQY,99
+lamin_cli-0.16.3.dist-info/METADATA,sha256=TMhHC2h49W_lpMqFg4TOu2JvdOxY9N4u2gJvNuDeO9E,336
+lamin_cli-0.16.3.dist-info/RECORD,,

lamin_cli-0.16.1.dist-info/RECORD

@@ -1,10 +0,0 @@
-lamin_cli/__init__.py,sha256=0yhPWneaCpI0YddQjW0b_76vC9Td3W2SiDUC16eBKlg,41
-lamin_cli/__main__.py,sha256=B2zUf8d7vcoCN-SuEo7dtlJg-waJgYyd-Y--ZZatP2E,7988
-lamin_cli/_cache.py,sha256=WMDUkJyATVdN8pMqgqw1DZLVym-CaBjCHEoRimI69LQ,813
-lamin_cli/_get.py,sha256=MXcR9PTW0FzY3uIPa-50SLCrtuRElxgDB3Vkfo-TFT0,1616
-lamin_cli/_migration.py,sha256=dULOvbpJ4VBiXuxPAM8jFGnBkh7pQGqE5eP-UC6uxWc,1055
-lamin_cli/_save.py,sha256=6mu6sAVcXsq4-XW-tzcVxUVh-Gv7e9EFkTyMmdrCGNw,4291
-lamin_cli-0.16.1.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
-lamin_cli-0.16.1.dist-info/WHEEL,sha256=Sgu64hAMa6g5FdzHxXv9Xdse9yxpGGMeagVtPMWpJQY,99
-lamin_cli-0.16.1.dist-info/METADATA,sha256=svDLkC_KV1xcYGK0H5GCWnQHVLm-h6x4P-NQnmZ6Owk,336
-lamin_cli-0.16.1.dist-info/RECORD,,