From 55859fe6671aa8db4375748aa12d99cc824430d0 Mon Sep 17 00:00:00 2001 From: Alex Wolf Date: Mon, 13 Oct 2025 21:41:11 +0200 Subject: [PATCH 1/2] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20Canonicalize=20import?= =?UTF-8?q?=20patterns?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docs/mlflow.ipynb | 5 +++-- docs/wandb.ipynb | 7 ++++--- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/docs/mlflow.ipynb b/docs/mlflow.ipynb index a8800d8..48720d3 100644 --- a/docs/mlflow.ipynb +++ b/docs/mlflow.ipynb @@ -50,8 +50,9 @@ }, "outputs": [], "source": [ - "import lamindb as ln\n", "import lightning as pl\n", + "import lamindb as ln\n", + "from lamindb.integrations import lightning as ll\n", "import mlflow\n", "from pathlib import Path\n", "\n", @@ -286,7 +287,7 @@ " autoencoder = LitAutoEncoder(hidden_size=32, bottleneck_size=16)\n", "\n", " # Create a LaminDB LightningCallback which also (optionally) annotates checkpoints by desired metrics\n", - " lamindb_callback = ln.integrations.lightning.Callback(\n", + " lamindb_callback = ll.Callback(\n", " path=Path(\"model_checkpoints\") / \"{mlflow_run.info.run_id}_last_epoch.ckpt\",\n", " key=f\"testmodels/mlflow/{mlflow_run.info.run_id}.ckpt\",\n", " features={\n", diff --git a/docs/wandb.ipynb b/docs/wandb.ipynb index 8c0d6b9..52f48df 100644 --- a/docs/wandb.ipynb +++ b/docs/wandb.ipynb @@ -46,9 +46,10 @@ }, "outputs": [], "source": [ + "import lightning as pl\n", "import lamindb as ln\n", + "from lamindb.integrations import lightning as ll\n", "import wandb\n", - "import lightning as pl\n", "from pathlib import Path\n", "\n", "from torch import utils\n", @@ -276,8 +277,8 @@ "source": [ "# Create a LaminDB LightningCallback which also (optionally) annotates checkpoints by desired metrics\n", "wandb_logger.experiment.id\n", - "lamindb_callback = ln.integrations.lightning.Callback(\n", - " path=Path(\"model_checkpoints\") / \"{wanddblogger.version}_last_epoch.ckpt\",\n", + "lamindb_callback = ll.Callback(\n", + " path=Path(\"model_checkpoints\") / f\"{wandb_logger.version}_last_epoch.ckpt\",\n", " key=f\"testmodels/wandb/{wandb_logger.experiment.id}.ckpt\",\n", " features={\n", " \"wandb_run_id\": wandb_logger.experiment.id,\n", From 540450c57dce3594672a71dd0d35b885a7fc9549 Mon Sep 17 00:00:00 2001 From: Alex Wolf Date: Mon, 13 Oct 2025 21:50:19 +0200 Subject: [PATCH 2/2] =?UTF-8?q?=F0=9F=A9=B9=20F-string?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docs/mlflow.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/mlflow.ipynb b/docs/mlflow.ipynb index 48720d3..59b587c 100644 --- a/docs/mlflow.ipynb +++ b/docs/mlflow.ipynb @@ -288,7 +288,7 @@ "\n", " # Create a LaminDB LightningCallback which also (optionally) annotates checkpoints by desired metrics\n", " lamindb_callback = ll.Callback(\n", - " path=Path(\"model_checkpoints\") / \"{mlflow_run.info.run_id}_last_epoch.ckpt\",\n", + " path=Path(\"model_checkpoints\") / f\"{mlflow_run.info.run_id}_last_epoch.ckpt\",\n", " key=f\"testmodels/mlflow/{mlflow_run.info.run_id}.ckpt\",\n", " features={\n", " \"mlflow_run_id\": mlflow_run.info.run_id,\n",