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

lamin_cli/__init__.py

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

lamin_cli/_get.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 from typing import Tuple
 from lamin_utils import logger
 import lamindb_setup as ln_setup
+from pathlib import Path
 
 
 def decompose_url(url: str) -> Tuple[str, str, str]:
@@ -25,6 +26,7 @@ def get(url: str):
     ln_setup.settings.auto_connect = False
 
     import lamindb as ln
+    from lamindb._finish import script_to_notebook
 
     ln_setup.settings.auto_connect = auto_connect
     ln.connect(instance_slug)
@@ -32,12 +34,23 @@ def get(url: str):
 
     if entity == "transform":
         transform = ln.Transform.get(uid)
-        filepath_cache = transform._source_code_artifact.cache()
         target_filename = transform.key
-        if not target_filename.endswith(transform._source_code_artifact.suffix):
-            target_filename += transform._source_code_artifact.suffix
-        filepath_cache.rename(target_filename)
-        logger.success(f"cached source code of transform {uid} as {target_filename}")
+        if transform._source_code_artifact_id is not None:
+            # backward compat
+            filepath_cache = transform._source_code_artifact.cache()
+            if not target_filename.endswith(transform._source_code_artifact.suffix):
+                target_filename += transform._source_code_artifact.suffix
+            filepath_cache.rename(target_filename)
+        elif transform.source_code is not None:
+            if transform.key.endswith(".ipynb"):
+                script_to_notebook(transform, target_filename)
+            else:
+                Path(target_filename).write_text(transform.source_code)
+        else:
+            raise ValueError("No source code available for this transform.")
+        logger.success(
+            f"downloaded source code of transform {uid} as {target_filename}"
+        )
     elif entity == "artifact":
         artifact = ln.Artifact.get(uid)
         cache_path = artifact.cache()

lamin_cli/_save.py

@@ -90,13 +90,10 @@ def save_from_filepath_cli(
                     " 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)
+            transform = ln.Transform.get(
+                uid__startswith=stem_uid, 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:
@@ -110,6 +107,5 @@ def save_from_filepath_cli(
             run=run,
             transform=transform,
             filepath=filepath,
-            transform_family=transform_family,
             from_cli=True,
         )

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

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

lamin_cli-0.17.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+lamin_cli/__init__.py,sha256=SHogvZAuS3pbUCg081hRt71yhRcoNP8eSHt63fGBZP0,41
+lamin_cli/__main__.py,sha256=7mUJe-6Vq8WNNR90t2j0Pa_XLMNaS9rdctj7OEfhum0,7898
+lamin_cli/_cache.py,sha256=6LrrWNrdZzH_BrohQ2n24xLFhUQ5EFan1bnzqMX_mIg,812
+lamin_cli/_get.py,sha256=wym5KQCsS02ttejFaVMlc2ZPK5SSnYNNbUqnj3nHSfg,2164
+lamin_cli/_migration.py,sha256=dULOvbpJ4VBiXuxPAM8jFGnBkh7pQGqE5eP-UC6uxWc,1055
+lamin_cli/_save.py,sha256=Wmio1l4QWpd2u_GIpmoeOukJujas2qtuB1BQfe7jyE4,4501
+lamin_cli-0.17.0.dist-info/entry_points.txt,sha256=Qms85i9cZPlu-U7RnVZhFsF7vJ9gaLZUFkCjcGcXTpg,49
+lamin_cli-0.17.0.dist-info/WHEEL,sha256=Sgu64hAMa6g5FdzHxXv9Xdse9yxpGGMeagVtPMWpJQY,99
+lamin_cli-0.17.0.dist-info/METADATA,sha256=x3-b49Ir0HiFQT-9G8N2unvmBPKJ42FasLPViwFQcvY,336
+lamin_cli-0.17.0.dist-info/RECORD,,

lamin_cli-0.16.3.dist-info/RECORD

@@ -1,10 +0,0 @@
-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,,