lamindb 0.76.8__py3-none-any.whl → 0.76.10__py3-none-any.whl

lamindb/_run.py

@@ -1,57 +1,60 @@
-from __future__ import annotations
-
-from lnschema_core.models import ParamManager, Run, Transform
-
-
-def __init__(run: Run, *args, **kwargs):
-    run.params = ParamManager(run)
-    if len(args) == len(run._meta.concrete_fields):
-        super(Run, run).__init__(*args, **kwargs)
-        return None
-    # now we proceed with the user-facing constructor
-    if len(args) > 1:
-        raise ValueError("Only one non-keyword arg allowed: transform")
-    transform: Transform = None
-    if "transform" in kwargs or len(args) == 1:
-        transform = kwargs.pop("transform") if len(args) == 0 else args[0]
-    reference: str | None = kwargs.pop("reference") if "reference" in kwargs else None
-    reference_type: str | None = (
-        kwargs.pop("reference_type") if "reference_type" in kwargs else None
-    )
-    if transform is None:
-        raise TypeError("Pass transform parameter")
-    if transform._state.adding:
-        raise ValueError("Please save transform record before creating a run")
-    super(Run, run).__init__(
-        transform=transform,
-        reference=reference,
-        reference_type=reference_type,
-    )
-
-
-def delete_run_artifacts(run: Run) -> None:
-    environment = None
-    if run.environment is not None:
-        environment = run.environment
-        run.environment = None
-    report = None
-    if run.report is not None:
-        report = run.report
-        run.report = None
-    if environment is not None or report is not None:
-        run.save()
-    if environment is not None:
-        # only delete if there are no other runs attached to this environment
-        if environment._environment_of.count() == 0:
-            environment.delete(permanent=True)
-    if report is not None:
-        report.delete(permanent=True)
-
-
-def delete(self) -> None:
-    delete_run_artifacts(self)
-    super(Run, self).delete()
-
-
-Run.__init__ = __init__
-Run.delete = delete
+from __future__ import annotations
+
+from lnschema_core.models import ParamManager, Run, Transform
+
+
+def __init__(run: Run, *args, **kwargs):
+    run.params = ParamManager(run)
+    if len(args) == len(run._meta.concrete_fields):
+        super(Run, run).__init__(*args, **kwargs)
+        return None
+    # now we proceed with the user-facing constructor
+    if len(args) > 1:
+        raise ValueError("Only one non-keyword arg allowed: transform")
+    transform: Transform = None
+    if "transform" in kwargs or len(args) == 1:
+        transform = kwargs.pop("transform") if len(args) == 0 else args[0]
+    reference: str | None = kwargs.pop("reference") if "reference" in kwargs else None
+    reference_type: str | None = (
+        kwargs.pop("reference_type") if "reference_type" in kwargs else None
+    )
+    parent: Run | None = kwargs.pop("parent", None)
+    if transform is None:
+        raise TypeError("Pass transform parameter")
+    if transform._state.adding:
+        raise ValueError("Please save transform record before creating a run")
+
+    super(Run, run).__init__(
+        transform=transform,
+        reference=reference,
+        parent=parent,
+        reference_type=reference_type,
+    )
+
+
+def delete_run_artifacts(run: Run) -> None:
+    environment = None
+    if run.environment is not None:
+        environment = run.environment
+        run.environment = None
+    report = None
+    if run.report is not None:
+        report = run.report
+        run.report = None
+    if environment is not None or report is not None:
+        run.save()
+    if environment is not None:
+        # only delete if there are no other runs attached to this environment
+        if environment._environment_of.count() == 0:
+            environment.delete(permanent=True)
+    if report is not None:
+        report.delete(permanent=True)
+
+
+def delete(self) -> None:
+    delete_run_artifacts(self)
+    super(Run, self).delete()
+
+
+Run.__init__ = __init__
+Run.delete = delete