hpcflow-new2 0.2.0a190__py3-none-any.whl → 0.2.0a200__py3-none-any.whl

hpcflow/sdk/persistence/types.py

@@ -1,13 +1,15 @@
 """
 Types used in type-checking the persistence subsystem.
 """
+
 from __future__ import annotations
 from typing import Any, Generic, TypeVar, TYPE_CHECKING
 from typing_extensions import TypedDict, NotRequired, TypeAlias
 
 if TYPE_CHECKING:
+    from collections.abc import Mapping
     from .base import StoreTask, StoreElement, StoreElementIter, StoreEAR, StoreParameter
-    from ..core.json_like import JSONDocument
+    from ..core.json_like import JSONed
     from ..core.parameters import ParameterValue
     from ..core.types import IterableParam
     from ..typing import DataIndex, ParamSource
@@ -65,6 +67,9 @@ class LoopDescriptor(TypedDict):
 
     #: The parameters iterated over by the loop.
     iterable_parameters: dict[str, IterableParam]
+    #: The parameters output by the loop, and the final task insert ID from which they
+    #: are output.
+    output_parameters: dict[str, int]
     #: The template data from which the loop was created.
     loop_template: NotRequired[dict[str, Any]]
     #: The number of iterations generated by a loop.
@@ -162,8 +167,12 @@ class RunMeta(TypedDict):
     run_hostname: str | None
     #: Whether the EAR succeeded, if known.
     success: bool | None
-    #: Whether the EAR was skipped.
-    skip: bool
+    #: The skip reason ID, if EAR was skipped.
+    skip: int
+    #: Port number used by ZeroMQ during execution.
+    port_number: int | None
+    #: Run ID whose command file can be used for this run (may be this run's ID).
+    commands_file_ID: int | None
 
 
 class TaskMeta(TypedDict):
@@ -276,6 +285,8 @@ class PersistenceCache(
     parameters: dict[int, AnySParameter]
     #: Number of element action runs.
     num_EARs: int | None
+    #: Number of parameters.
+    num_params: int | None
 
 
 class ZarrAttrsDict(TypedDict):
@@ -302,6 +313,6 @@ class ZarrAttrsDict(TypedDict):
     #: Loops in the workflow.
     loops: list[dict[str, Any]]
     #: Submissions by the workflow.
-    submissions: list[JSONDocument]
+    submissions: list[Mapping[str, JSONed]]
     #: Replacement workflow, if any.
     replaced_workflow: NotRequired[str]