lamindb_setup 0.78.0__tar.gz → 0.79.0__tar.gz

{lamindb_setup-0.78.0 → lamindb_setup-0.79.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lamindb_setup
-Version: 0.78.0
+Version: 0.79.0
 Summary: Setup & configure LaminDB.
 Author-email: Lamin Labs <open-source@lamin.ai>
 Requires-Python: >=3.9

{lamindb_setup-0.78.0 → lamindb_setup-0.79.0}/docs/hub-cloud/02-connect-local-instance.ipynb

@@ -85,29 +85,6 @@
     ")"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "0986a5f3",
-   "metadata": {},
-   "source": [
-    "You cannot load another instance in the same Python session:"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "bc3568ae",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import pytest\n",
-    "\n",
-    "with pytest.raises(RuntimeError):\n",
-    "    ln_setup.connect(\"testuser2/mydata\")\n",
-    "\n",
-    "assert ln_setup.settings.instance.slug == \"testuser1/mydata\""
-   ]
-  },
   {
    "cell_type": "markdown",
    "id": "e3701120",

{lamindb_setup-0.78.0 → lamindb_setup-0.79.0}/docs/hub-cloud/08-test-multi-session.ipynb

@@ -113,14 +113,18 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "with pytest.raises(RuntimeError):\n",
+    "from lamindb_setup._init_instance import CannotSwitchDefaultInstance\n",
+    "\n",
+    "with pytest.raises(CannotSwitchDefaultInstance):\n",
     "    ln_setup.init(storage=\"./testsetup2\")\n",
-    "with pytest.raises(RuntimeError):\n",
+    "with pytest.raises(CannotSwitchDefaultInstance):\n",
     "    ln_setup.connect(\"testsetup\")\n",
     "with pytest.raises(RuntimeError):\n",
     "    ln_setup.migrate.create()\n",
     "with pytest.raises(RuntimeError):\n",
-    "    ln_setup.migrate.deploy()"
+    "    ln_setup.migrate.deploy()\n",
+    "\n",
+    "assert ln_setup.settings.instance.slug == \"testuser1/testsetup\""
    ]
   },
   {

{lamindb_setup-0.78.0 → lamindb_setup-0.79.0}/lamindb_setup/__init__.py

@@ -33,7 +33,7 @@ Modules & settings:
 
 """
 
-__version__ = "0.78.0"  # denote a release candidate for 0.1.0 with 0.1rc1
+__version__ = "0.79.0"  # denote a release candidate for 0.1.0 with 0.1rc1
 
 import os as _os
 import sys as _sys

{lamindb_setup-0.78.0 → lamindb_setup-0.79.0}/lamindb_setup/_connect_instance.py

@@ -9,7 +9,11 @@ from lamin_utils import logger
 
 from ._check_setup import _check_instance_setup
 from ._close import close as close_instance
-from ._init_instance import MESSAGE_NO_MULTIPLE_INSTANCE, load_from_isettings
+from ._init_instance import (
+    MESSAGE_CANNOT_SWITCH_DEFAULT_INSTANCE,
+    CannotSwitchDefaultInstance,
+    load_from_isettings,
+)
 from ._silence_loggers import silence_loggers
 from .core._hub_core import connect_instance_hub
 from .core._hub_utils import (
@@ -224,7 +228,9 @@ def connect(slug: str, **kwargs) -> str | tuple | None:
                 logger.info(f"connected lamindb: {settings.instance.slug}")
                 return None
             else:
-                raise RuntimeError(MESSAGE_NO_MULTIPLE_INSTANCE)
+                raise CannotSwitchDefaultInstance(
+                    MESSAGE_CANNOT_SWITCH_DEFAULT_INSTANCE
+                )
         elif (
             _write_settings
             and settings._instance_exists

{lamindb_setup-0.78.0 → lamindb_setup-0.79.0}/lamindb_setup/_init_instance.py

@@ -26,7 +26,7 @@ if TYPE_CHECKING:
     from .core.types import UPathStr
 
 
-def get_schema_module_name(schema_name) -> str:
+def get_schema_module_name(schema_name, raise_import_error: bool = True) -> str | None:
     import importlib.util
 
     name_attempts = [f"lnschema_{schema_name.replace('-', '_')}", schema_name]
@@ -34,9 +34,11 @@ def get_schema_module_name(schema_name) -> str:
         module_spec = importlib.util.find_spec(name)
         if module_spec is not None:
             return name
-    raise ImportError(
-        f"Python package for '{schema_name}' is not installed.\nIf your package is on PyPI, run `pip install {schema_name}`"
-    )
+    message = f"Schema module '{schema_name}' is not installed → no access to its labels & registries (resolve via `pip install {schema_name}`)"
+    if raise_import_error:
+        raise ImportError(message)
+    logger.warning(message.lower())
+    return None
 
 
 def register_storage_in_instance(ssettings: StorageSettings):
@@ -180,6 +182,8 @@ def validate_init_args(
         validate_schema_arg,
     )
 
+    if storage is None:
+        raise SystemExit("✗ `storage` argument can't be `None`")
     # should be called as the first thing
     name_str = infer_instance_name(storage=storage, name=name, db=db)
     owner_str = settings.user.handle if _user is None else _user.handle
@@ -206,10 +210,18 @@ def validate_init_args(
     return name_str, instance_id, instance_state, instance_slug
 
 
-MESSAGE_NO_MULTIPLE_INSTANCE = """
-Currently don't support subsequent connection to different databases in the same
-Python session.\n
-Try running on the CLI: lamin settings set auto-connect false
+class CannotSwitchDefaultInstance(SystemExit):
+    pass
+
+
+MESSAGE_CANNOT_SWITCH_DEFAULT_INSTANCE = """
+You cannot write to different instances in the same Python session.
+
+Do you want to read from another instance via `Record.using()`? For example:
+
+ln.Artifact.using("laminlabs/cellxgene").filter()
+
+Or do you want to switch off auto-connect via `lamin settings set auto-connect false`?
 """
 
 
@@ -248,7 +260,7 @@ def init(
         from ._check_setup import _check_instance_setup
 
         if _check_instance_setup() and not _test:
-            raise RuntimeError(MESSAGE_NO_MULTIPLE_INSTANCE)
+            raise CannotSwitchDefaultInstance(MESSAGE_CANNOT_SWITCH_DEFAULT_INSTANCE)
         elif _write_settings:
             close_instance(mute=True)
         from .core._hub_core import init_instance as init_instance_hub

{lamindb_setup-0.78.0 → lamindb_setup-0.79.0}/lamindb_setup/core/_settings_storage.py

@@ -85,8 +85,8 @@ def init_storage(
     StorageSettings,
     Literal["hub-record-not-created", "hub-record-retireved", "hub-record-created"],
 ]:
-    if root is None:
-        raise ValueError("`storage` argument can't be `None`")
+    assert root is not None, "`root` argument can't be `None`"
+
     root_str = str(root)  # ensure we have a string
     if ".lamindb" in root_str:
         raise ValueError(

{lamindb_setup-0.78.0 → lamindb_setup-0.79.0}/lamindb_setup/core/django.py

@@ -55,7 +55,17 @@ def setup_django(
         from .._init_instance import get_schema_module_name
 
         schema_names = ["core"] + list(isettings.schema)
-        installed_apps = [get_schema_module_name(n) for n in schema_names]
+        raise_import_error = True if init else False
+        installed_apps = [
+            package_name
+            for n in schema_names
+            if (
+                package_name := get_schema_module_name(
+                    n, raise_import_error=raise_import_error
+                )
+            )
+            is not None
+        ]
         if view_schema:
             installed_apps = installed_apps[::-1]  # to fix how apps appear
             installed_apps += ["schema_graph", "django.contrib.staticfiles"]

{lamindb_setup-0.78.0 → lamindb_setup-0.79.0}/tests/hub-cloud/test_edge_request.py

@@ -43,7 +43,7 @@ def test_edge_request():
         assert instance["owner"] == "laminlabs"
         assert instance["name"] == "lamindata"
         assert instance["api_url"] == "https://us-east-1.api.lamin.ai"
-        assert instance["schema_str"] == "bionty,wetlab"
+        assert instance["schema_str"].startswith("bionty,wetlab")
         assert "schema_id" in instance
         # check that schema_id is well-formed UUID
         # if not, throws ValueError

{lamindb_setup-0.78.0 → lamindb_setup-0.79.0}/tests/hub-cloud/test_init_instance.py

@@ -32,6 +32,12 @@ def test_init_instance_postgres_default_name(get_hub_client):
         ln_setup.delete(instance_name, force=True)
     except InstanceNotFoundError:
         pass
+    # test init with storage=None
+    # this happens when calling CLI lamin init or lamin init --name smth
+    with pytest.raises(SystemExit):
+        ln_setup.init(storage=None, _test=True)
+    with pytest.raises(SystemExit):
+        ln_setup.init(storage=None, name="init-to-fail", _test=True)
     # now, run init
     ln_setup.init(storage="./mydatapg", db=pgurl, _test=True)
     assert ln_setup.settings.instance.slug == "testuser2/pgtest"