lamindb 0.68.0__py3-none-any.whl → 0.68.2__py3-none-any.whl

lamindb/__init__.py

@@ -39,6 +39,7 @@ Functions:
 .. autosummary::
    :toctree: .
 
+   connect
    track
    view
    save
@@ -54,41 +55,23 @@ Modules & settings:
 
 """
 
-__version__ = "0.68.0"  # denote a release candidate for 0.1.0 with 0.1rc1
+__version__ = "0.68.2"  # denote a release candidate for 0.1.0 with 0.1rc1
 
 import os as _os
 
 import lamindb_setup as _lamindb_setup
+from lamindb_setup._check_setup import InstanceNotSetupError, _check_instance_setup
+from lamindb_setup._connect_instance import connect
+from lamindb_setup.core.upath import UPath
 
-# prints warning of python versions
-from lamin_utils import py_version_warning as _py_version_warning
-from lamindb_setup import _check_instance_setup
-from lamindb_setup._check_instance_setup import _INSTANCE_NOT_SETUP_WARNING
-from lamindb_setup._init_instance import reload_schema_modules as _reload_schema_modules
-from lamindb_setup.dev.upath import UPath
-
-_py_version_warning("3.8", "3.11")
-
-_TESTING = _lamindb_setup._TESTING
-_INSTANCE_SETUP = _check_instance_setup(from_lamindb=True)
-
-# allow the user to call setup
 from . import setup
 
 
-class InstanceNotSetupError(Exception):
-    pass
-
-
 def __getattr__(name):
-    raise InstanceNotSetupError(
-        f"{_INSTANCE_NOT_SETUP_WARNING}If you used the CLI to init or load an instance,"
-        " please RESTART the python session (in a notebook, restart kernel)"
-    )
+    raise InstanceNotSetupError()
 
 
-# only import all other functionality if setup was successful
-if _INSTANCE_SETUP:
+if _check_instance_setup(from_lamindb=True):
     del InstanceNotSetupError
     del __getattr__  # delete so that imports work out
     from lnschema_core.models import (
@@ -128,6 +111,10 @@ if _INSTANCE_SETUP:
 
     # schema modules
     if not _os.environ.get("LAMINDB_MULTI_INSTANCE") == "true":
+        from lamindb_setup._init_instance import (
+            reload_schema_modules as _reload_schema_modules,
+        )
+
         _reload_schema_modules(_lamindb_setup.settings.instance)
 
     track = run_context._track

lamindb/_artifact.py

@@ -2,26 +2,33 @@ from pathlib import Path, PurePath, PurePosixPath
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 import fsspec
+import lamindb_setup as ln_setup
 import pandas as pd
 from anndata import AnnData
 from lamin_utils import colors, logger
 from lamindb_setup import settings as setup_settings
 from lamindb_setup._init_instance import register_storage
-from lamindb_setup.dev import StorageSettings
-from lamindb_setup.dev._docs import doc_args
-from lamindb_setup.dev.upath import create_path, extract_suffix_from_path
+from lamindb_setup.core import StorageSettings
+from lamindb_setup.core._docs import doc_args
+from lamindb_setup.core.hashing import b16_to_b64, hash_file, hash_md5s_from_dir
+from lamindb_setup.core.types import UPathStr
+from lamindb_setup.core.upath import (
+    create_path,
+    extract_suffix_from_path,
+    get_stat_dir_gs,
+    get_stat_dir_s3,
+    get_stat_file_cloud,
+)
 from lnschema_core import Artifact, Run, Storage
 from lnschema_core.models import IsTree
 from lnschema_core.types import (
     DataLike,
-    PathLike,
     VisibilityChoice,
 )
 
 from lamindb._utils import attach_func_to_class_method
 from lamindb.core._data import _track_run_input
 from lamindb.core._settings import settings
-from lamindb.core.hashing import b16_to_b64, hash_file, hash_md5s_from_dir
 from lamindb.core.storage import (
     LocalPathClasses,
     UPath,
@@ -38,7 +45,6 @@ from lamindb.core.storage.file import (
 )
 from lamindb.core.versioning import get_uid_from_old_version, init_uid
 
-from . import _TESTING
 from .core._data import (
     add_transform_to_kwargs,
     get_run,
@@ -106,7 +112,7 @@ def process_pathlike(
 
 def process_data(
     provisional_uid: str,
-    data: Union[PathLike, DataLike],
+    data: Union[UPathStr, DataLike],
     format: Optional[str],
     key: Optional[str],
     default_storage: Storage,
@@ -115,7 +121,7 @@ def process_data(
 ) -> Tuple[Any, Union[Path, UPath], str, Storage, bool]:
     """Serialize a data object that's provided as file or in memory."""
     # if not overwritten, data gets stored in default storage
-    if isinstance(data, (str, Path, UPath)):  # PathLike, spelled out
+    if isinstance(data, (str, Path, UPath)):  # UPathStr, spelled out
         access_token = (
             default_storage._access_token
             if hasattr(default_storage, "_access_token")
@@ -162,69 +168,6 @@ def process_data(
     return memory_rep, path, suffix, storage, use_existing_storage_key
 
 
-def get_stat_file_cloud(stat: Dict) -> Tuple[int, str, str]:
-    size = stat["size"]
-    # small files
-    if "-" not in stat["ETag"]:
-        # only store hash for non-multipart uploads
-        # we can't rapidly validate multi-part uploaded files client-side
-        # we can add more logic later down-the-road
-        hash = b16_to_b64(stat["ETag"])
-        hash_type = "md5"
-    else:
-        stripped_etag, suffix = stat["ETag"].split("-")
-        suffix = suffix.strip('"')
-        hash = f"{b16_to_b64(stripped_etag)}-{suffix}"
-        hash_type = "md5-n"  # this is the S3 chunk-hashing strategy
-    return size, hash, hash_type
-
-
-def get_stat_dir_s3(path: UPath) -> Tuple[int, str, str, int]:
-    import boto3
-    from lamindb_setup.dev.upath import AWS_CREDENTIALS_PRESENT
-
-    if not AWS_CREDENTIALS_PRESENT:
-        # passing the following param directly to Session() doesn't
-        # work, unfortunately: botocore_session=path.fs.session
-        from botocore import UNSIGNED
-        from botocore.config import Config
-
-        config = Config(signature_version=UNSIGNED)
-        s3 = boto3.session.Session().resource("s3", config=config)
-    else:
-        s3 = boto3.session.Session().resource("s3")
-    bucket, key, _ = path.fs.split_path(path.as_posix())
-    # assuming this here is the fastest way of querying for many objects
-    objects = s3.Bucket(bucket).objects.filter(Prefix=key)
-    size = sum([object.size for object in objects])
-    md5s = [
-        # skip leading and trailing quotes
-        object.e_tag[1:-1]
-        for object in objects
-    ]
-    n_objects = len(md5s)
-    hash, hash_type = hash_md5s_from_dir(md5s)
-    return size, hash, hash_type, n_objects
-
-
-def get_stat_dir_gs(path: UPath) -> Tuple[int, str, str, int]:
-    import google.cloud.storage as gc_storage
-
-    bucket, key, _ = path.fs.split_path(path.as_posix())
-    # assuming this here is the fastest way of querying for many objects
-    client = gc_storage.Client(
-        credentials=path.fs.credentials.credentials, project=path.fs.project
-    )
-    objects = client.Bucket(bucket).list_blobs(prefix=key)
-    sizes, md5s = [], []
-    for object in objects:
-        sizes.append(object.size)
-        md5s.append(object.md5_hash)
-    n_objects = len(md5s)
-    hash, hash_type = hash_md5s_from_dir(md5s)
-    return sum(sizes), hash, hash_type, n_objects
-
-
 def get_stat_or_artifact(
     path: UPath,
     suffix: str,
@@ -690,7 +633,7 @@ def from_anndata(
 @doc_args(Artifact.from_dir.__doc__)
 def from_dir(
     cls,
-    path: PathLike,
+    path: UPathStr,
     key: Optional[str] = None,
     *,
     run: Optional[Run] = None,
@@ -791,7 +734,7 @@ def from_dir(
 # docstring handled through attach_func_to_class_method
 def replace(
     self,
-    data: Union[PathLike, DataLike],
+    data: Union[UPathStr, DataLike],
     run: Optional[Run] = None,
     format: Optional[str] = None,
 ) -> None:
@@ -906,19 +849,21 @@ def delete(
     storage: Optional[bool] = None,
     using_key: Optional[str] = None,
 ) -> None:
-    # by default, we only move artifacts into the trash
-    if self.visibility > -1 and permanent is not True:
+    # by default, we only move artifacts into the trash (visibility = -1)
+    trash_visibility = VisibilityChoice.trash.value
+    if self.visibility > trash_visibility and permanent is not True:
         if storage is not None:
             logger.warning("moving artifact to trash, storage arg is ignored")
         # move to trash
-        self.visibility = -1
+        self.visibility = trash_visibility
         self.save()
-        logger.warning("moved artifact to trash (visibility = -1)")
+        logger.warning(f"moved artifact to trash (visibility = {trash_visibility})")
         return
 
     # if the artifact is already in the trash
     # permanent delete skips the trash
     if permanent is None:
+        # ask for confirmation of permanent delete
         response = input(
             "Artifact record is already in trash! Are you sure you want to permanently"
             " delete it? (y/n) You can't undo this action."
@@ -928,13 +873,14 @@ def delete(
         # this second option doesn't feel very intuitive
         delete_record = permanent
 
-    if delete_record:
+    if delete_record is True:
         # need to grab file path before deletion
         filepath = filepath_from_artifact(self, using_key)
         # only delete in storage if DB delete is successful
         # DB delete might error because of a foreign key constraint violated etc.
         self._delete_skip_storage()
         if self.key is None or self.key_is_virtual:
+            # always delete in storage if the key is virtual
             delete_in_storage = True
             if storage is not None:
                 logger.warning("storage arg is ignored if storage key is non-semantic")
@@ -951,9 +897,10 @@ def delete(
                 delete_in_storage = storage
         # we don't yet have logic to bring back the deleted metadata record
         # in case storage deletion fails - this is important for ACID down the road
-        if delete_in_storage:
-            delete_storage(filepath)
-            logger.success(f"deleted {colors.yellow(f'{filepath}')}")
+        if delete_in_storage is True:
+            delete_msg = delete_storage(filepath)
+            if delete_msg != "did-not-delete":
+                logger.success(f"deleted {colors.yellow(f'{filepath}')}")
 
 
 def _delete_skip_storage(artifact, *args, **kwargs) -> None:
@@ -1036,7 +983,7 @@ METHOD_NAMES = [
     "view_tree",
 ]
 
-if _TESTING:
+if ln_setup._TESTING:
     from inspect import signature
 
     SIGS = {

lamindb/_collection.py

@@ -12,10 +12,12 @@ from typing import (
 )
 
 import anndata as ad
+import lamindb_setup as ln_setup
 import pandas as pd
 from anndata import AnnData
 from lamin_utils import logger
-from lamindb_setup.dev._docs import doc_args
+from lamindb_setup.core._docs import doc_args
+from lamindb_setup.core.hashing import hash_set
 from lnschema_core.models import Collection, CollectionArtifact, FeatureSet
 from lnschema_core.types import DataLike, VisibilityChoice
 
@@ -24,7 +26,7 @@ from lamindb.core._data import _track_run_input
 from lamindb.core._mapped_collection import MappedCollection
 from lamindb.core.versioning import get_uid_from_old_version, init_uid
 
-from . import _TESTING, Artifact, Run
+from . import Artifact, Run
 from ._artifact import data_is_anndata
 from ._query_set import QuerySet
 from ._registry import init_self_from_db
@@ -34,7 +36,6 @@ from .core._data import (
     save_feature_set_links,
     save_feature_sets,
 )
-from .core.hashing import hash_set
 
 if TYPE_CHECKING:
     from lamindb.core.storage._backed_access import AnnDataAccessor, BackedAccessor
@@ -408,10 +409,11 @@ def load(
 # docstring handled through attach_func_to_class_method
 def delete(self, permanent: Optional[bool] = None) -> None:
     # change visibility to trash
-    if self.visibility > VisibilityChoice.trash.value and permanent is not True:
-        self.visibility = VisibilityChoice.trash.value
+    trash_visibility = VisibilityChoice.trash.value
+    if self.visibility > trash_visibility and permanent is not True:
+        self.visibility = trash_visibility
         self.save()
-        logger.warning("moved collection to trash.")
+        logger.warning(f"moved collection to trash (visibility = {trash_visibility})")
         return
 
     # permanent delete
@@ -481,7 +483,7 @@ METHOD_NAMES = [
     "restore",
 ]
 
-if _TESTING:
+if ln_setup._TESTING:
     from inspect import signature
 
     SIGS = {

lamindb/_feature.py

@@ -1,7 +1,8 @@
 from typing import Dict, List, Optional, Union
 
+import lamindb_setup as ln_setup
 import pandas as pd
-from lamindb_setup.dev._docs import doc_args
+from lamindb_setup.core._docs import doc_args
 from lnschema_core.models import Feature, Registry
 from lnschema_core.types import FieldAttr
 from pandas.api.types import CategoricalDtype, is_string_dtype
@@ -9,7 +10,6 @@ from pandas.api.types import CategoricalDtype, is_string_dtype
 from lamindb._utils import attach_func_to_class_method
 from lamindb.core._settings import settings
 
-from . import _TESTING
 from ._query_set import RecordsList
 
 FEATURE_TYPES = {
@@ -181,7 +181,7 @@ METHOD_NAMES = [
     "save",
 ]
 
-if _TESTING:
+if ln_setup._TESTING:
     from inspect import signature
 
     SIGS = {

lamindb/_feature_set.py

@@ -1,15 +1,15 @@
 from typing import TYPE_CHECKING, Dict, Iterable, List, Optional, Type, Union
 
+import lamindb_setup as ln_setup
 import numpy as np
 from lamin_utils import logger
-from lamindb_setup.dev._docs import doc_args
+from lamindb_setup.core._docs import doc_args
+from lamindb_setup.core.hashing import hash_set
 from lnschema_core import Feature, FeatureSet, Registry, ids
 from lnschema_core.types import FieldAttr, ListLike
 
 from lamindb._utils import attach_func_to_class_method
-from lamindb.core.hashing import hash_set
 
-from . import _TESTING
 from ._feature import convert_numpy_dtype_to_lamin_feature_type
 from ._registry import init_self_from_db
 
@@ -250,7 +250,7 @@ METHOD_NAMES = [
     "save",
 ]
 
-if _TESTING:
+if ln_setup._TESTING:
     from inspect import signature
 
     SIGS = {

lamindb/_is_versioned.py

@@ -1,12 +1,12 @@
 from typing import Optional
 
+import lamindb_setup as ln_setup
 from lamin_utils import logger
-from lamindb_setup.dev.upath import UPath
+from lamindb_setup.core.upath import UPath
 from lnschema_core.models import IsVersioned
 
 from lamindb._utils import attach_func_to_class_method
 
-from . import _TESTING
 from .core.versioning import get_new_path_from_uid, get_uid_from_old_version
 
 
@@ -33,7 +33,7 @@ METHOD_NAMES = [
     "add_to_version_family",
 ]
 
-if _TESTING:  # type: ignore
+if ln_setup._TESTING:  # type: ignore
     from inspect import signature
 
     SIGS = {name: signature(getattr(IsVersioned, name)) for name in METHOD_NAMES}

lamindb/_parents.py

@@ -1,13 +1,13 @@
 import builtins
 from typing import List, Optional, Set, Union
 
+import lamindb_setup as ln_setup
 from lamin_utils import logger
 from lnschema_core import Artifact, Collection, Registry, Run, Transform
 from lnschema_core.models import HasParents, format_field_value
 
 from lamindb._utils import attach_func_to_class_method
 
-from . import _TESTING
 from ._registry import StrField, get_default_str_field
 
 LAMIN_GREEN_LIGHTER = "#10b981"
@@ -427,7 +427,7 @@ METHOD_NAMES = [
     "view_parents",
 ]
 
-if _TESTING:  # type: ignore
+if ln_setup._TESTING:  # type: ignore
     from inspect import signature
 
     SIGS = {

lamindb/_query_manager.py

@@ -2,7 +2,7 @@ from typing import NamedTuple, Optional
 
 from django.db import models
 from lamin_utils import logger
-from lamindb_setup.dev._docs import doc_args
+from lamindb_setup.core._docs import doc_args
 from lnschema_core.models import Registry
 from lnschema_core.types import StrField
 

lamindb/_query_set.py

@@ -3,7 +3,7 @@ from typing import Dict, Iterable, List, NamedTuple, Optional, Union
 
 import pandas as pd
 from django.db import models
-from lamindb_setup.dev._docs import doc_args
+from lamindb_setup.core._docs import doc_args
 from lnschema_core.models import (
     Artifact,
     CanValidate,

lamindb/_registry.py

@@ -11,18 +11,17 @@ from django.db.models import Manager, QuerySet
 from lamin_utils import logger
 from lamin_utils._lookup import Lookup
 from lamin_utils._search import search as base_search
+from lamindb_setup._connect_instance import get_owner_name_from_identifier
 from lamindb_setup._init_instance import InstanceSettings
-from lamindb_setup._load_instance import get_owner_name_from_identifier
-from lamindb_setup.dev._docs import doc_args
-from lamindb_setup.dev._hub_core import load_instance
-from lamindb_setup.dev._settings_storage import StorageSettings
+from lamindb_setup.core._docs import doc_args
+from lamindb_setup.core._hub_core import connect_instance
+from lamindb_setup.core._settings_storage import StorageSettings
 from lnschema_core import Registry
 from lnschema_core.types import ListLike, StrField
 
 from lamindb._utils import attach_func_to_class_method
 from lamindb.core._settings import settings
 
-from . import _TESTING
 from ._from_values import get_or_create_records
 
 IPYTHON = getattr(builtins, "__IPYTHON__", False)
@@ -356,16 +355,16 @@ def using(
     instance: str,
 ) -> "QuerySet":
     """{}."""
-    from lamindb_setup._load_instance import (
+    from lamindb_setup._connect_instance import (
         load_instance_settings,
         update_db_using_local,
     )
-    from lamindb_setup.dev._settings_store import instance_settings_file
+    from lamindb_setup.core._settings_store import instance_settings_file
 
     owner, name = get_owner_name_from_identifier(instance)
     settings_file = instance_settings_file(name, owner)
     if not settings_file.exists():
-        load_result = load_instance(owner=owner, name=name)
+        load_result = connect_instance(owner=owner, name=name)
         if isinstance(load_result, str):
             raise RuntimeError(
                 f"Failed to load instance {instance}, please check your permission!"
@@ -413,18 +412,21 @@ def update_fk_to_default_db(
             setattr(records, f"{fk}_id", fk_record_default.id)
 
 
+FKBULK = [
+    "organism",
+    "public_source",
+    "initial_version",
+    "latest_report",  # Transform
+    "source_code",  # Transform
+    "report",  # Run
+    "file",  # Collection
+]
+
+
 def transfer_fk_to_default_db_bulk(
     records: Union[List, QuerySet], using_key: Optional[str]
 ):
-    for fk in [
-        "organism",
-        "public_source",
-        "initial_version",
-        "latest_report",  # Transform
-        "source_code",  # Transform
-        "report",  # Run
-        "file",  # Collection
-    ]:
+    for fk in FKBULK:
         update_fk_to_default_db(records, fk, using_key)
 
 
@@ -433,6 +435,7 @@ def transfer_to_default_db(
     using_key: Optional[str],
     save: bool = False,
     mute: bool = False,
+    transfer_fk: bool = True,
 ) -> Optional[Registry]:
     db = record._state.db
     if db is not None and db != "default" and using_key is None:
@@ -459,14 +462,24 @@ def transfer_to_default_db(
             else:
                 logger.warning(WARNING_RUN_TRANSFORM)
                 record.run_id = None
-        if hasattr(record, "transform_id"):
+        if hasattr(record, "transform_id") and record._meta.model_name != "run":
             record.transform = None
             if run_context.transform is not None:
                 record.transform_id = run_context.transform.id
             else:
                 record.transform_id = None
-        update_fk_to_default_db(record, "storage", using_key)
-        update_fk_to_default_db(record, "artifact", using_key)
+        # transfer other foreign key fields
+        fk_fields = [
+            i.name
+            for i in record._meta.fields
+            if i.get_internal_type() == "ForeignKey"
+            if i.name not in {"created_by", "run", "transform"}
+        ]
+        if not transfer_fk:
+            # don't transfer fk fields that are already bulk transferred
+            fk_fields = [fk for fk in fk_fields if fk not in FKBULK]
+        for fk in fk_fields:
+            update_fk_to_default_db(record, fk, using_key)
         record.id = None
         record._state.db = "default"
         if save:
@@ -485,15 +498,19 @@ def save(self, *args, **kwargs) -> None:
     if self.__class__.__name__ == "Collection" and self.id is not None:
         # when creating a new collection without being able to access artifacts
         artifacts = self.artifacts.list()
+    # transfer of the record to the default db with fk fields
     result = transfer_to_default_db(self, using_key)
     if result is not None:
         init_self_from_db(self, result)
     else:
         # here, we can't use the parents argument
+        # parents are not saved for the self record
         save_kwargs = kwargs.copy()
         if "parents" in save_kwargs:
             save_kwargs.pop("parents")
         super(Registry, self).save(*args, **save_kwargs)
+    # perform transfer of many-to-many fields
+    # only supported for Artifact and Collection records
     if db is not None and db != "default" and using_key is None:
         if self.__class__.__name__ == "Collection":
             if len(artifacts) > 0:
@@ -507,6 +524,7 @@ def save(self, *args, **kwargs) -> None:
             self_on_db = copy(self)
             self_on_db._state.db = db
             self_on_db.pk = pk_on_db
+            # by default, transfer parents of the labels to maintain ontological hierarchy
             add_from_kwargs = {"parents": kwargs.get("parents", True)}
             logger.info("transfer features")
             self.features._add_from(self_on_db, **add_from_kwargs)
@@ -523,7 +541,7 @@ METHOD_NAMES = [
     "using",
 ]
 
-if _TESTING:  # type: ignore
+if ln_setup._TESTING:  # type: ignore
     from inspect import signature
 
     SIGS = {

lamindb/_save.py

@@ -10,7 +10,7 @@ import lamindb_setup
 from django.db import transaction
 from django.utils.functional import partition
 from lamin_utils import logger
-from lamindb_setup.dev.upath import print_hook
+from lamindb_setup.core.upath import print_hook
 from lnschema_core.models import Artifact, Registry
 
 from lamindb.core._settings import settings

lamindb/_storage.py

@@ -1,5 +1,5 @@
-from lamindb_setup.dev._docs import doc_args
-from lamindb_setup.dev.upath import UPath, create_path
+from lamindb_setup.core._docs import doc_args
+from lamindb_setup.core.upath import UPath, create_path
 from lnschema_core import Storage
 
 

lamindb/_transform.py

@@ -1,13 +1,11 @@
-from typing import TYPE_CHECKING, Optional
+from typing import Optional
 
-from lnschema_core.models import TRANSFORM_TYPE_DEFAULT, Artifact, Run, Transform
+from lnschema_core.models import Artifact, Run, Transform
+from lnschema_core.types import TransformType
 
 from ._run import delete_run_artifacts
 from .core.versioning import get_uid_from_old_version, init_uid
 
-if TYPE_CHECKING:
-    from lnschema_core.types import TransformType
-
 
 def __init__(transform: Transform, *args, **kwargs):
     if len(args) == len(transform._meta.concrete_fields):
@@ -23,7 +21,7 @@ def __init__(transform: Transform, *args, **kwargs):
     (kwargs.pop("initial_version_id") if "initial_version_id" in kwargs else None)
     version: Optional[str] = kwargs.pop("version") if "version" in kwargs else None
     type: Optional[TransformType] = (
-        kwargs.pop("type") if "type" in kwargs else TRANSFORM_TYPE_DEFAULT
+        kwargs.pop("type") if "type" in kwargs else TransformType.pipeline
     )
     reference: Optional[str] = (
         kwargs.pop("reference") if "reference" in kwargs else None
@@ -85,5 +83,11 @@ def delete(self) -> None:
     super(Transform, self).delete()
 
 
+@property  # type: ignore
+def latest_run(self) -> Run:
+    return self.runs.order_by("-run_at").first()
+
+
 Transform.__init__ = __init__
 Transform.delete = delete
+Transform.latest_run = latest_run

lamindb/_ulabel.py

@@ -1,12 +1,12 @@
 from typing import List, Optional
 
-from lamindb_setup.dev._docs import doc_args
+import lamindb_setup as ln_setup
+from lamindb_setup.core._docs import doc_args
 from lnschema_core import ULabel
 from lnschema_core.types import ListLike
 
 from lamindb._utils import attach_func_to_class_method
 
-from . import _TESTING
 from ._from_values import get_or_create_records
 
 
@@ -55,7 +55,7 @@ METHOD_NAMES = [
     "from_values",
 ]
 
-if _TESTING:
+if ln_setup._TESTING:
     from inspect import signature
 
     SIGS = {