lamindb 0.64.1__py3-none-any.whl → 0.64.2__py3-none-any.whl

lamindb/__init__.py

@@ -54,7 +54,7 @@ Modules & settings:
 
 """
 
-__version__ = "0.64.1"  # denote a release candidate for 0.1.0 with 0.1rc1
+__version__ = "0.64.2"  # denote a release candidate for 0.1.0 with 0.1rc1
 
 import os as _os
 
@@ -72,7 +72,7 @@ _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  # noqa
+from . import setup
 
 
 class InstanceNotSetupError(Exception):
@@ -90,7 +90,7 @@ def __getattr__(name):
 if _INSTANCE_SETUP:
     del InstanceNotSetupError
     del __getattr__  # delete so that imports work out
-    from lnschema_core import (  # noqa
+    from lnschema_core import (
         Artifact,
         Dataset,
         Feature,
@@ -104,25 +104,25 @@ if _INSTANCE_SETUP:
 
     File = Artifact  # backward compat
     from . import _artifact  # noqa
-    from . import _dataset  # noqa
-    from . import _feature  # noqa
-    from . import _feature_set  # noqa
-    from . import _parents  # noqa
-    from . import _registry  # noqa
-    from . import _run  # noqa
-    from . import _storage  # noqa
-    from . import _transform  # noqa
-    from . import _ulabel  # noqa
-    from . import _validate  # noqa
-    from . import dev  # noqa
-    from ._delete import delete  # noqa
-    from ._save import save  # noqa
-    from ._view import view  # noqa
-    from .dev._run_context import run_context  # noqa
+    from . import _dataset
+    from . import _feature
+    from . import _feature_set
+    from . import _parents
+    from . import _registry
+    from . import _run
+    from . import _storage
+    from . import _transform
+    from . import _ulabel
+    from . import _validate
+    from . import dev
+    from ._delete import delete
+    from ._save import save
+    from ._view import view
+    from .dev._run_context import run_context
     from .dev._settings import settings
 
     # schema modules
     _reload_schema_modules(_lamindb_setup.settings.instance)
 
-    track = run_context._track  # noqa
+    track = run_context._track
     settings.__doc__ = """Global :class:`~lamindb.dev.Settings`."""

lamindb/_artifact.py

@@ -1,5 +1,5 @@
 from pathlib import Path, PurePath, PurePosixPath
-from typing import Any, Dict, List, Optional, Tuple, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 import anndata as ad
 import fsspec
@@ -36,7 +36,6 @@ from lamindb.dev.storage import (
     size_adata,
     write_to_file,
 )
-from lamindb.dev.storage._backed_access import AnnDataAccessor, BackedAccessor
 from lamindb.dev.storage.file import (
     auto_storage_key_from_artifact,
     auto_storage_key_from_artifact_uid,
@@ -54,6 +53,9 @@ from .dev._data import (
 )
 from .dev.storage.file import AUTO_KEY_PREFIX
 
+if TYPE_CHECKING:
+    from lamindb.dev.storage._backed_access import AnnDataAccessor, BackedAccessor
+
 
 def process_pathlike(
     filepath: UPath, skip_existence_check: bool = False
@@ -405,34 +407,34 @@ def get_artifact_kwargs_from_data(
     if check_path_in_storage:
         key_is_virtual = False
 
-    kwargs = dict(
-        suffix=suffix,
-        hash=hash,
-        hash_type=hash_type,
-        key=key,
-        size=size,
-        storage_id=storage.id,
+    kwargs = {
+        "suffix": suffix,
+        "hash": hash,
+        "hash_type": hash_type,
+        "key": key,
+        "size": size,
+        "storage_id": storage.id,
         # passing both the id and the object
         # to make them both available immediately
         # after object creation
-        n_objects=n_objects,
-        n_observations=None,  # to implement
-        run_id=run.id if run is not None else None,
-        run=run,
-        key_is_virtual=key_is_virtual,
-    )
+        "n_objects": n_objects,
+        "n_observations": None,  # to implement
+        "run_id": run.id if run is not None else None,
+        "run": run,
+        "key_is_virtual": key_is_virtual,
+    }
     if not isinstance(path, LocalPathClasses):
         local_filepath = None
         cloud_filepath = path
     else:
         local_filepath = path
         cloud_filepath = None
-    privates = dict(
-        local_filepath=local_filepath,
-        cloud_filepath=cloud_filepath,
-        memory_rep=memory_rep,
-        check_path_in_storage=check_path_in_storage,
-    )
+    privates = {
+        "local_filepath": local_filepath,
+        "cloud_filepath": cloud_filepath,
+        "memory_rep": memory_rep,
+        "check_path_in_storage": check_path_in_storage,
+    }
     return kwargs, privates
 
 
@@ -623,7 +625,7 @@ def from_df(
     is_new_version_of: Optional["Artifact"] = None,
     **kwargs,
 ) -> "Artifact":
-    """{}"""
+    """{}."""
     artifact = Artifact(
         data=df,
         key=key,
@@ -698,7 +700,7 @@ def from_anndata(
     is_new_version_of: Optional["Artifact"] = None,
     **kwargs,
 ) -> "Artifact":
-    """{}"""
+    """{}."""
     artifact = Artifact(
         data=adata,
         key=key,
@@ -721,7 +723,7 @@ def from_dir(
     *,
     run: Optional[Run] = None,
 ) -> List["Artifact"]:
-    """{}"""
+    """{}."""
     logger.warning(
         "this creates one artifact per file in the directory - you might simply call"
         " ln.Artifact(dir) to get one artifact for the entire directory"
@@ -788,7 +790,7 @@ def from_dir(
             non_unique_artifacts = {
                 hash: artifact
                 for hash, artifact in artifacts_dict.items()
-                if artifact.hash in seen_hashes or seen_hashes.add(artifact.hash)  # type: ignore  # noqa
+                if artifact.hash in seen_hashes or seen_hashes.add(artifact.hash)  # type: ignore
             }
             display_non_unique = "\n    ".join(
                 f"{artifact}" for artifact in non_unique_artifacts
@@ -995,7 +997,7 @@ def _save_skip_storage(file, *args, **kwargs) -> None:
 @property  # type: ignore
 @doc_args(Artifact.path.__doc__)
 def path(self) -> Union[Path, UPath]:
-    """{}"""
+    """{}."""
     return filepath_from_artifact(self)
 
 
@@ -1008,7 +1010,7 @@ def view_tree(
     length_limit: int = 1000,
     max_files_per_dir_per_type: int = 7,
 ) -> None:
-    """{}"""
+    """{}."""
     from lamindb.dev._view_tree import view_tree as _view_tree
 
     _view_tree(
@@ -1056,6 +1058,6 @@ for name in METHOD_NAMES:
 # privates currently dealt with separately
 Artifact._delete_skip_storage = _delete_skip_storage
 Artifact._save_skip_storage = _save_skip_storage
-setattr(Artifact, "path", path)
+Artifact.path = path
 # this seems a Django-generated function
 delattr(Artifact, "get_visibility_display")

lamindb/_dataset.py

@@ -1,5 +1,5 @@
 from collections import defaultdict
-from typing import Dict, Iterable, List, Literal, Optional, Tuple, Union
+from typing import TYPE_CHECKING, Dict, Iterable, List, Literal, Optional, Tuple, Union
 
 import anndata as ad
 import pandas as pd
@@ -11,7 +11,6 @@ from lnschema_core.types import AnnDataLike, DataLike, FieldAttr, VisibilityChoi
 from lamindb._utils import attach_func_to_class_method
 from lamindb.dev._data import _track_run_input
 from lamindb.dev._mapped_dataset import MappedDataset
-from lamindb.dev.storage._backed_access import AnnDataAccessor, BackedAccessor
 from lamindb.dev.versioning import get_ids_from_old_version, init_uid
 
 from . import _TESTING, Artifact, Run
@@ -25,6 +24,9 @@ from .dev._data import (
 )
 from .dev.hashing import hash_set
 
+if TYPE_CHECKING:
+    from lamindb.dev.storage._backed_access import AnnDataAccessor, BackedAccessor
+
 
 def __init__(
     dataset: Dataset,
@@ -69,7 +71,7 @@ def __init__(
     )
     if not len(kwargs) == 0:
         raise ValueError(
-            f"Only data, name, run, description, reference, reference_type, visibility can be passed, you passed: {kwargs}"  # noqa
+            f"Only data, name, run, description, reference, reference_type, visibility can be passed, you passed: {kwargs}"
         )
 
     if is_new_version_of is None:
@@ -198,7 +200,7 @@ def from_df(
     is_new_version_of: Optional["Artifact"] = None,
     **kwargs,
 ) -> "Dataset":
-    """{}"""
+    """{}."""
     feature_set = FeatureSet.from_df(df, field=field, **kwargs)
     if feature_set is not None:
         feature_sets = {"columns": feature_set}
@@ -233,7 +235,7 @@ def from_anndata(
     is_new_version_of: Optional["Artifact"] = None,
     **kwargs,
 ) -> "Dataset":
-    """{}"""
+    """{}."""
     if isinstance(adata, Artifact):
         assert not adata._state.adding
         assert adata.accessor == "AnnData"
@@ -259,7 +261,7 @@ def from_anndata(
 def from_artifacts(artifacts: Iterable[Artifact]) -> Tuple[str, Dict[str, str]]:
     # assert all artifacts are already saved
     logger.debug("check not saved")
-    saved = not any([artifact._state.adding for artifact in artifacts])
+    saved = not any(artifact._state.adding for artifact in artifacts)
     if not saved:
         raise ValueError("Not all artifacts are yet saved, please save them")
     # query all feature sets of artifacts

lamindb/_delete.py

@@ -1,4 +1,4 @@
-from typing import List, Union, overload  # noqa
+from typing import List, Union, overload
 
 from lamin_utils import colors, logger
 from lnschema_core import Registry

lamindb/_feature.py

@@ -36,7 +36,7 @@ def __init__(self, *args, **kwargs):
     # now we proceed with the user-facing constructor
     if len(args) != 0:
         raise ValueError("Only non-keyword args allowed")
-    type: Optional[Union[type, str]] = kwargs.pop("type") if "type" in kwargs else None
+    type: Optional[Union[type, str]] = kwargs.pop("type") if "type" in kwargs else None  # noqa: F823
     registries: Optional[List[Registry]] = (
         kwargs.pop("registries") if "registries" in kwargs else None
     )
@@ -87,7 +87,7 @@ def categoricals_from_df(df: "pd.DataFrame") -> Dict:
 @classmethod  # type:ignore
 @doc_args(Feature.from_df.__doc__)
 def from_df(cls, df: "pd.DataFrame") -> List["Feature"]:
-    """{}"""
+    """{}."""
     categoricals = categoricals_from_df(df)
 
     types = {}
@@ -146,7 +146,7 @@ def from_df(cls, df: "pd.DataFrame") -> List["Feature"]:
 
 @doc_args(Feature.save.__doc__)
 def save(self, *args, **kwargs) -> None:
-    """{}"""
+    """{}."""
     super(Feature, self).save(*args, **kwargs)
 
 

lamindb/_feature_set.py

@@ -1,7 +1,6 @@
-from typing import Dict, Iterable, List, Optional, Type, Union
+from typing import TYPE_CHECKING, Dict, Iterable, List, Optional, Type, Union
 
 import numpy as np
-import pandas as pd
 from lamin_utils import logger
 from lamindb_setup.dev._docs import doc_args
 from lnschema_core import Feature, FeatureSet, Registry, ids
@@ -12,9 +11,13 @@ from lamindb.dev.hashing import hash_set
 
 from . import _TESTING
 from ._feature import convert_numpy_dtype_to_lamin_feature_type
-from ._query_set import QuerySet
 from ._registry import init_self_from_db
 
+if TYPE_CHECKING:
+    import pandas as pd
+
+    from ._query_set import QuerySet
+
 NUMBER_TYPE = "number"
 
 
@@ -75,14 +78,16 @@ def validate_features(features: List[Registry]) -> Registry:
         if len(features) == 0:
             raise ValueError("Provide list of features with at least one element")
     except TypeError:
-        raise ValueError("Please pass a ListLike of features, not a single feature")
+        raise ValueError(
+            "Please pass a ListLike of features, not a single feature"
+        ) from None
     if not hasattr(features, "__getitem__"):
         raise TypeError("features has to be list-like")
     if not isinstance(features[0], Registry):
         raise TypeError(
             "features has to store feature records! use .from_values() otherwise"
         )
-    feature_types = set([feature.__class__ for feature in features])
+    feature_types = {feature.__class__ for feature in features}
     if len(feature_types) > 1:
         raise TypeError("feature_set can only contain a single type")
     for feature in features:
@@ -130,7 +135,7 @@ def __init__(self, *args, **kwargs):
 
 @doc_args(FeatureSet.save.__doc__)
 def save(self, *args, **kwargs) -> None:
-    """{}"""
+    """{}."""
     super(FeatureSet, self).save(*args, **kwargs)
     if hasattr(self, "_features"):
         related_name, records = self._features
@@ -157,7 +162,7 @@ def from_values(
     name: Optional[str] = None,
     **kwargs,
 ) -> Optional["FeatureSet"]:
-    """{}"""
+    """{}."""
     if not isinstance(field, FieldAttr):
         raise TypeError(
             "Argument `field` must be a Registry field, e.g., `Feature.name`"
@@ -191,7 +196,7 @@ def from_df(
     name: Optional[str] = None,
     **kwargs,
 ) -> Optional["FeatureSet"]:
-    """{}"""
+    """{}."""
     registry = field.field.model
     validated = registry.validate(df.columns, field=field, **kwargs)
     if validated.sum() == 0:
@@ -219,7 +224,7 @@ def from_df(
 @property  # type: ignore
 @doc_args(FeatureSet.members.__doc__)
 def members(self) -> "QuerySet":
-    """{}"""
+    """{}."""
     if self._state.adding:
         # this should return a queryset and not a list...
         # need to fix this
@@ -255,5 +260,5 @@ if _TESTING:
 for name in METHOD_NAMES:
     attach_func_to_class_method(name, FeatureSet, globals())
 
-setattr(FeatureSet, "members", members)
-setattr(FeatureSet, "_get_related_name", _get_related_name)
+FeatureSet.members = members
+FeatureSet._get_related_name = _get_related_name

lamindb/_filter.py

@@ -12,7 +12,7 @@ def filter(Registry: Type[Registry], **expressions) -> QuerySet:
         # visibility is set to 0 unless expressions contains id or uid equality
         if not ("id" in expressions or "uid" in expressions):
             visibility = "visibility"
-            if not any([e.startswith(visibility) for e in expressions]):
+            if not any(e.startswith(visibility) for e in expressions):
                 expressions[
                     visibility
                 ] = VisibilityChoice.default.value  # default visibility

lamindb/_from_values.py

@@ -53,7 +53,7 @@ def get_or_create_records(
                 n_nonval = colors.yellow(f"{len(unmapped_values)} non-validated")
                 logger.warning(
                     f"{colors.red('did not create')} {name} record{s} for "
-                    f"{n_nonval} {colors.italic(f'{field.field.name}{s}')}: {print_values}"  # noqa
+                    f"{n_nonval} {colors.italic(f'{field.field.name}{s}')}: {print_values}"
                 )
         if Registry.__module__.startswith("lnschema_bionty.") or Registry == ULabel:
             if isinstance(iterable, pd.Series):
@@ -74,8 +74,10 @@ def get_or_create_records(
 def get_existing_records(
     iterable_idx: pd.Index,
     field: StrField,
-    kwargs: Dict = {},
+    kwargs: Dict = None,
 ):
+    if kwargs is None:
+        kwargs = {}
     model = field.field.model
     condition: Dict = {} if len(kwargs) == 0 else kwargs.copy()
     # existing records matching is agnostic to the bionty source
@@ -194,7 +196,7 @@ def create_records_from_bionty(
         print_values = colors.purple(_print_values(names))
         msg_syn = (
             "created"
-            f" {colors.purple(f'{len(syn_mapper)} {model.__name__} record{s} from Bionty')}"  # noqa
+            f" {colors.purple(f'{len(syn_mapper)} {model.__name__} record{s} from Bionty')}"
             f" matching {colors.italic('synonyms')}: {print_values}"
         )
 
@@ -221,11 +223,9 @@ def create_records_from_bionty(
             if len(msg) > 0:
                 logger.success(msg)
             logger.success(
-                (
-                    "created"
-                    f" {colors.purple(f'{len(validated)} {model.__name__} record{s} from Bionty')}"  # noqa
-                    f" matching {colors.italic(f'{field.field.name}')}: {print_values}"  # noqa
-                )
+                "created"
+                f" {colors.purple(f'{len(validated)} {model.__name__} record{s} from Bionty')}"
+                f" matching {colors.italic(f'{field.field.name}')}: {print_values}"
             )
 
     # make sure that synonyms logging appears after the field logging

lamindb/_parents.py

@@ -61,15 +61,15 @@ def view_parents(
     )
 
 
-def view_flow(data: Union[Artifact, Dataset], with_children: bool = True) -> None:
+def view_lineage(data: Union[Artifact, Dataset], with_children: bool = True) -> None:
     """Graph of data flow.
 
     Notes:
         For more info, see use cases: :doc:`docs:data-flow`.
 
     Examples:
-        >>> dataset.view_flow()
-        >>> artifact.view_flow()
+        >>> dataset.view_lineage()
+        >>> artifact.view_lineage()
     """
     import graphviz
 
@@ -276,16 +276,16 @@ def _record_label(record: Registry, field: Optional[str] = None):
     elif isinstance(record, Run):
         name = f'{record.transform.name.replace("&", "&")}'
         return (
-            rf'<{TRANSFORM_EMOJIS.get(str(record.transform.type), "💫")} {name}<BR/><FONT COLOR="GREY" POINT-SIZE="10"'  # noqa
-            rf' FACE="Monospace">uid={record.transform.uid}<BR/>type={record.transform.type},'  # noqa
-            rf" user={record.created_by.name}<BR/>run_at={format_field_value(record.run_at)}</FONT>>"  # noqa
+            rf'<{TRANSFORM_EMOJIS.get(str(record.transform.type), "💫")} {name}<BR/><FONT COLOR="GREY" POINT-SIZE="10"'
+            rf' FACE="Monospace">uid={record.transform.uid}<BR/>type={record.transform.type},'
+            rf" user={record.created_by.name}<BR/>run_at={format_field_value(record.run_at)}</FONT>>"
         )
     elif isinstance(record, Transform):
         name = f'{record.name.replace("&", "&amp;")}'
         return (
-            rf'<{TRANSFORM_EMOJIS.get(str(record.type), "💫")} {name}<BR/><FONT COLOR="GREY" POINT-SIZE="10"'  # noqa
+            rf'<{TRANSFORM_EMOJIS.get(str(record.type), "💫")} {name}<BR/><FONT COLOR="GREY" POINT-SIZE="10"'
             rf' FACE="Monospace">uid={record.uid}<BR/>type={record.type},'
-            rf" user={record.created_by.name}<BR/>updated_at={format_field_value(record.updated_at)}</FONT>>"  # noqa
+            rf" user={record.created_by.name}<BR/>updated_at={format_field_value(record.updated_at)}</FONT>>"
         )
     else:
         name = record.__getattribute__(field)

lamindb/_query_manager.py

@@ -59,9 +59,9 @@ class QueryManager(models.Manager):
         """
         self._track_run_input_manager()
         if field is None:
-            return [item for item in self.all()]
+            return list(self.all())
         else:
-            return [item for item in self.values_list(field, flat=True)]
+            return list(self.values_list(field, flat=True))
 
     def df(self, **kwargs):
         """Convert to DataFrame.
@@ -80,14 +80,14 @@ class QueryManager(models.Manager):
 
     @doc_args(Registry.search.__doc__)
     def search(self, string: str, **kwargs):
-        """{}"""
+        """{}."""
         from ._registry import _search
 
         return _search(cls=self.all(), string=string, **kwargs)
 
     @doc_args(Registry.lookup.__doc__)
     def lookup(self, field: Optional[StrField] = None, **kwargs) -> NamedTuple:
-        """{}"""
+        """{}."""
         from ._registry import _lookup
 
         return _lookup(cls=self.all(), field=field, **kwargs)
@@ -107,15 +107,13 @@ class QueryManager(models.Manager):
             return
 
 
-setattr(models.Manager, "list", QueryManager.list)
-setattr(models.Manager, "df", QueryManager.df)
-setattr(models.Manager, "search", QueryManager.search)
-setattr(models.Manager, "lookup", QueryManager.lookup)
-setattr(models.Manager, "__getitem__", QueryManager.__getitem__)
-setattr(
-    models.Manager, "_track_run_input_manager", QueryManager._track_run_input_manager
-)
+models.Manager.list = QueryManager.list
+models.Manager.df = QueryManager.df
+models.Manager.search = QueryManager.search
+models.Manager.lookup = QueryManager.lookup
+models.Manager.__getitem__ = QueryManager.__getitem__
+models.Manager._track_run_input_manager = QueryManager._track_run_input_manager
 # the two lines below would be easy if we could actually inherit; like this,
 # they're suboptimal
-setattr(models.Manager, "all_base_class", models.Manager.all)
-setattr(models.Manager, "all", QueryManager.all)
+models.Manager.all_base_class = models.Manager.all
+models.Manager.all = QueryManager.all

lamindb/_query_set.py

@@ -52,7 +52,7 @@ class QuerySet(models.QuerySet, CanValidate, IsTree):
         Examples:
 
             >>> ln.save(ln.ULabel.from_values(["ULabel1", "ULabel2", "ULabel3"], field="name")) # noqa
-            >>> ln.ULabel.filter().df()
+            >>> ln.ULabel.df()
             >>> label = ln.ULabel.filter(name="ULabel1").one()
             >>> label = ln.ULabel.filter(name="benchmark").one()
             >>> label.parents.add(label)
@@ -144,9 +144,9 @@ class QuerySet(models.QuerySet, CanValidate, IsTree):
             ['ULabel1', 'ULabel2', 'ULabel3']
         """
         if field is None:
-            return [item for item in self]
+            return list(self)
         else:
-            return [item for item in self.values_list(field, flat=True)]
+            return list(self.values_list(field, flat=True))
 
     def first(self) -> Optional[Registry]:
         """If non-empty, the first result in the query set, otherwise None.
@@ -195,14 +195,14 @@ class QuerySet(models.QuerySet, CanValidate, IsTree):
 
     @doc_args(Registry.search.__doc__)
     def search(self, string: str, **kwargs):
-        """{}"""
+        """{}."""
         from ._registry import _search
 
         return _search(cls=self, string=string, **kwargs)
 
     @doc_args(Registry.lookup.__doc__)
     def lookup(self, field: Optional[StrField] = None, **kwargs) -> NamedTuple:
-        """{}"""
+        """{}."""
         from ._registry import _lookup
 
         return _lookup(cls=self, field=field, **kwargs)
@@ -211,7 +211,7 @@ class QuerySet(models.QuerySet, CanValidate, IsTree):
     def validate(
         self, values: ListLike, field: Optional[Union[str, StrField]] = None, **kwargs
     ):
-        """{}"""
+        """{}."""
         from ._validate import _validate
 
         return _validate(cls=self, values=values, field=field, **kwargs)
@@ -220,7 +220,7 @@ class QuerySet(models.QuerySet, CanValidate, IsTree):
     def inspect(
         self, values: ListLike, field: Optional[Union[str, StrField]] = None, **kwargs
     ):
-        """{}"""
+        """{}."""
         from ._validate import _inspect
 
         return _inspect(cls=self, values=values, field=field, **kwargs)
@@ -229,7 +229,7 @@ class QuerySet(models.QuerySet, CanValidate, IsTree):
     def standardize(
         self, values: Iterable, field: Optional[Union[str, StrField]] = None, **kwargs
     ):
-        """{}"""
+        """{}."""
         from ._validate import _standardize
 
         return _standardize(cls=self, values=values, field=field, **kwargs)
@@ -242,7 +242,7 @@ class QuerySet(models.QuerySet, CanValidate, IsTree):
         length_limit: int = 1000,
         max_files_per_dir_per_type: int = 7,
     ) -> None:
-        """{}"""
+        """{}."""
         from .dev._view_tree import view_tree as _view_tree
 
         _view_tree(
@@ -254,13 +254,13 @@ class QuerySet(models.QuerySet, CanValidate, IsTree):
         )
 
 
-setattr(models.QuerySet, "df", QuerySet.df)
-setattr(models.QuerySet, "list", QuerySet.list)
-setattr(models.QuerySet, "first", QuerySet.first)
-setattr(models.QuerySet, "one", QuerySet.one)
-setattr(models.QuerySet, "one_or_none", QuerySet.one_or_none)
-setattr(models.QuerySet, "search", QuerySet.search)
-setattr(models.QuerySet, "lookup", QuerySet.lookup)
-setattr(models.QuerySet, "validate", QuerySet.validate)
-setattr(models.QuerySet, "inspect", QuerySet.inspect)
-setattr(models.QuerySet, "standardize", QuerySet.standardize)
+models.QuerySet.df = QuerySet.df
+models.QuerySet.list = QuerySet.list
+models.QuerySet.first = QuerySet.first
+models.QuerySet.one = QuerySet.one
+models.QuerySet.one_or_none = QuerySet.one_or_none
+models.QuerySet.search = QuerySet.search
+models.QuerySet.lookup = QuerySet.lookup
+models.QuerySet.validate = QuerySet.validate
+models.QuerySet.inspect = QuerySet.inspect
+models.QuerySet.standardize = QuerySet.standardize