sequence-base 0.4.29__tar.gz → 0.4.30__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {sequence_base-0.4.29 → sequence_base-0.4.30}/PKG-INFO +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/pyproject.toml +1 -1
- sequence_base-0.4.30/sequence_base/by_name.py +90 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/constants/version.py +2 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/construct.py +26 -5
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/policy.py +14 -7
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ActRequest.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ActResponse.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ActionChunk.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ActionContract.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ActionStep.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Actions.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/AdapterPair.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/AddLocal.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Autoscaler.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/BenchmarkConditionSpec.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/BenchmarkRegisterRequest.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/BenchmarkSpec.json +2 -2
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/BenchmarkStatusResponse.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/BootstrapResponse.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Camera.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/CodeBundle.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Concurrency.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Conditions.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ConnectRequest.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ConnectResponse.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/DeploymentAccepted.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/DeploymentPolicy.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/DeploymentRequest.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/DeploymentStatusResponse.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Embodiment.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ErrorBody.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ErrorCodes.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Estimate.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/EvalDryRunResponse.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/EvalReport.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/EvalRequest.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/EvalResponse.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ImageFrame.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ImageSpec.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ImageStepAddLocal.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ImageStepApt.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ImageStepEnv.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ImageStepProject.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ImageStepRun.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ImageStepTemplateFiles.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ImageStepUvPip.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/InferParam.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Lease.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/MetricAgg.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Observation.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ObservationContract.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Observations.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ObserveRequest.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ObserveResponse.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/PolicyRegistration.json +6 -5
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/PolicySpec.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Project.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Proprioception.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/RenderSpec.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/ReplaySpec.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Resources.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/RolloutResult.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/SecretCreateRequest.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/SecretInfo.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/SecretRef.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/SeqFlags.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Step.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Stream.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/TaskAgg.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/TemplateFiles.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Tensors.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Timeouts.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/UploadTarget.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/Variant.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/VolumeFile.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/VolumeInfo.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/VolumeMount.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/VolumeRef.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/VolumeUploadRequest.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/VolumeUploadResponse.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schemas/WarmingError.json +1 -1
- {sequence_base-0.4.29 → sequence_base-0.4.30}/.gitignore +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/README.md +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/__init__.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/auth.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/codec/__init__.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/codec/image.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/constants/__init__.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/constants/cards.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/constants/headers.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/constants/limits.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/__init__.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/accounts.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/act.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/adapters.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/benchmarks.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/connect.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/deployments.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/errors.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/eval.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/flags.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/io.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/obs_schedule.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/observe.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/own_io.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/templates.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/volumes.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/contract/wire.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/episode.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/errors/__init__.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/errors/transport.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/feedback.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/msgpack_io.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/obs/__init__.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/obs/log.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/obs/rid.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/presets/__init__.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/presets/frames.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/presets/robot_arm.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/schema_export.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/telemetry/__init__.py +0 -0
- {sequence_base-0.4.29 → sequence_base-0.4.30}/sequence_base/telemetry/events.py +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.5
|
|
2
2
|
Name: sequence-base
|
|
3
|
-
Version: 0.4.
|
|
3
|
+
Version: 0.4.30
|
|
4
4
|
Summary: Shared foundation for the Sequences platform — wire contract, correlation, error taxonomy, constants. Used by the robot SDK, gateway, and container harness.
|
|
5
5
|
Project-URL: Changelog, https://generalsequences.com/docs/changelog/
|
|
6
6
|
Requires-Python: >=3.10
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
[project]
|
|
2
2
|
name = "sequence-base"
|
|
3
|
-
version = "0.4.
|
|
3
|
+
version = "0.4.30"
|
|
4
4
|
description = "Shared foundation for the Sequences platform — wire contract, correlation, error taxonomy, constants. Used by the robot SDK, gateway, and container harness."
|
|
5
5
|
requires-python = ">=3.10"
|
|
6
6
|
dependencies = ["pydantic>=2.0"]
|
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
"""
|
|
2
|
+
Which arguments of a method a policy class names (infer=, observe=) an act's inputs fill: the inputs whole, its one
|
|
3
|
+
argument - or, when it takes two or more without a default, each key of the inputs to its argument of that name.
|
|
4
|
+
|
|
5
|
+
In one sentence: `takes_by_name(parameters, platform)` reads the method's arguments - less *args, **kwargs and the ones
|
|
6
|
+
the platform hands it itself (`platform`: INFER_PLATFORM, OBSERVE_PLATFORM) → two or more of them without a default:
|
|
7
|
+
it takes its inputs by name (ByName), each of its arguments a key - one without a default a key the inputs must hold,
|
|
8
|
+
one with a default a key they may → `ByName.call(inputs, what)` makes what it is called with, refusing inputs that are
|
|
9
|
+
no dict, lack a key it needs, or hold one it takes no argument for (unless it takes **kwargs), each named; fewer than
|
|
10
|
+
two: None, the inputs go to its first argument whole and its others with a default are inference parameters, as before.
|
|
11
|
+
|
|
12
|
+
WHY by the arguments without a default: a method that needs two or more could never be handed an act's inputs whole -
|
|
13
|
+
the platform refused to deploy one - so binding them by name changes nothing a class deployed today does, and a method
|
|
14
|
+
that takes one beside its inference parameters keeps taking its inputs whole. The SDK reads the rule when a class is
|
|
15
|
+
declared (seq/named.py, registry.infer_params), the worker when it calls the method (harness/handler/loader.py): one
|
|
16
|
+
rule on both sides.
|
|
17
|
+
"""
|
|
18
|
+
from __future__ import annotations
|
|
19
|
+
|
|
20
|
+
import inspect
|
|
21
|
+
from collections.abc import Iterable, Mapping
|
|
22
|
+
from dataclasses import dataclass
|
|
23
|
+
from typing import Any
|
|
24
|
+
|
|
25
|
+
__all__ = ["INFER_PLATFORM", "OBSERVE_PLATFORM", "ByName", "takes_by_name"]
|
|
26
|
+
|
|
27
|
+
|
|
28
|
+
|
|
29
|
+
INFER_PLATFORM = frozenset({"seed", "episode_id", "episode", "feedback", "plan", "robot", "embodiment"})
|
|
30
|
+
OBSERVE_PLATFORM = frozenset({"step"})
|
|
31
|
+
|
|
32
|
+
_VARIADIC = (inspect.Parameter.VAR_POSITIONAL, inspect.Parameter.VAR_KEYWORD)
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
@dataclass(frozen=True)
|
|
36
|
+
class ByName:
|
|
37
|
+
"""A method that takes an act's inputs by name: `names`, the arguments they fill, in order; `required`, those with
|
|
38
|
+
no default; `positional`, those it takes by position only (before a `/`); `rest`, whether it takes **kwargs - any
|
|
39
|
+
other key of the inputs going there; `platform`, the names the platform hands it itself."""
|
|
40
|
+
|
|
41
|
+
names: tuple[str, ...]
|
|
42
|
+
required: tuple[str, ...]
|
|
43
|
+
positional: tuple[str, ...]
|
|
44
|
+
rest: bool
|
|
45
|
+
platform: frozenset[str]
|
|
46
|
+
|
|
47
|
+
def call(self, inputs: Any, what: str) -> tuple[list, dict]:
|
|
48
|
+
"""(args, kwargs) the method `what` ("Model.predict") is called with for an act's `inputs`; ValueError, in its
|
|
49
|
+
words, for inputs that are not a dict of its arguments."""
|
|
50
|
+
takes = ", ".join(n if n in self.required else f"{n} (optional)" for n in self.names)
|
|
51
|
+
if not isinstance(inputs, Mapping):
|
|
52
|
+
raise ValueError(f"{what} takes its inputs by name ({takes}): an act's inputs are a dict of them, and "
|
|
53
|
+
f"these are a {type(inputs).__name__}")
|
|
54
|
+
missing = [n for n in self.required if n not in inputs]
|
|
55
|
+
extra = [k for k in inputs if k not in self.names
|
|
56
|
+
and not (self.rest and isinstance(k, str) and k not in self.platform)]
|
|
57
|
+
if missing or extra:
|
|
58
|
+
held = f"lack {_keys(missing)}" if missing else ""
|
|
59
|
+
if extra:
|
|
60
|
+
held += (" and " if held else "") + f"hold {_keys(extra)}, which it takes no argument for"
|
|
61
|
+
ours = [k for k in extra if k in self.platform]
|
|
62
|
+
mine = f" ({_keys(ours)}: the platform's to hand it)" if ours else ""
|
|
63
|
+
raise ValueError(f"{what} takes its inputs by name ({takes}), and these {held}{mine}")
|
|
64
|
+
args: list = []
|
|
65
|
+
for i, name in enumerate(self.positional):
|
|
66
|
+
if name not in inputs:
|
|
67
|
+
later = next((n for n in self.positional[i + 1:] if n in inputs), None)
|
|
68
|
+
if later is not None:
|
|
69
|
+
raise ValueError(f"{what} takes {name} and {later} by position only, in that order: these hold "
|
|
70
|
+
f"{later!r} and no {name!r}")
|
|
71
|
+
break
|
|
72
|
+
args.append(inputs[name])
|
|
73
|
+
return args, {k: v for k, v in inputs.items() if k not in self.positional}
|
|
74
|
+
|
|
75
|
+
|
|
76
|
+
def takes_by_name(parameters: Iterable[inspect.Parameter], platform: frozenset[str] = frozenset()) -> ByName | None:
|
|
77
|
+
"""How a method of these `parameters` (self left out) takes an act's inputs: by name (ByName) when two or more of
|
|
78
|
+
its arguments - *args, **kwargs and `platform`'s left out - have no default, else None: whole, as one argument."""
|
|
79
|
+
every = list(parameters)
|
|
80
|
+
params = [p for p in every if p.kind not in _VARIADIC and p.name not in platform]
|
|
81
|
+
required = tuple(p.name for p in params if p.default is inspect.Parameter.empty)
|
|
82
|
+
if len(required) < 2:
|
|
83
|
+
return None
|
|
84
|
+
return ByName(names=tuple(p.name for p in params), required=required,
|
|
85
|
+
positional=tuple(p.name for p in params if p.kind is inspect.Parameter.POSITIONAL_ONLY),
|
|
86
|
+
rest=any(p.kind is inspect.Parameter.VAR_KEYWORD for p in every), platform=frozenset(platform))
|
|
87
|
+
|
|
88
|
+
|
|
89
|
+
def _keys(names: list) -> str:
|
|
90
|
+
return ", ".join(repr(n) for n in names)
|
|
@@ -1,10 +1,12 @@
|
|
|
1
1
|
"""
|
|
2
2
|
How a policy class that names its own methods is built (PolicyRegistration.constructor): its constructor - or the
|
|
3
|
-
factory it names - called with the arguments its decorator was given, once
|
|
3
|
+
factory it names, a classmethod of it or a module's function - called with the arguments its decorator was given, once
|
|
4
|
+
two forms in them are filled in.
|
|
4
5
|
|
|
5
6
|
In one sentence: a value {"$config": path, "$set": {...}} becomes the YAML or JSON file at `path` - from the folder of
|
|
6
7
|
the class's file - read where the policy runs, with `$set`'s keys laid over its top level → in every string,
|
|
7
8
|
"{weights}" becomes the weights directory and "{weights.<part>}" a named part's → `cls.<make>(*args, **kwargs)`, or
|
|
9
|
+
`function(*args, **kwargs)` for make="package.module:function" - the module imported where the policy runs - or
|
|
8
10
|
`cls(*args, **kwargs)` without a factory.
|
|
9
11
|
|
|
10
12
|
WHY a config file is read where the policy runs: it is part of the project (XPolicyLab's deploy.yml beside its
|
|
@@ -14,12 +16,14 @@ worker's directory.
|
|
|
14
16
|
"""
|
|
15
17
|
from __future__ import annotations
|
|
16
18
|
|
|
19
|
+
import importlib
|
|
17
20
|
import json
|
|
18
21
|
import os
|
|
19
22
|
import re
|
|
20
23
|
from typing import Any
|
|
21
24
|
|
|
22
|
-
__all__ = ["CONFIG_KEY", "SET_KEY", "ConstructError", "build", "config", "expand", "is_config", "
|
|
25
|
+
__all__ = ["CONFIG_KEY", "SET_KEY", "ConstructError", "build", "config", "expand", "is_config", "maker",
|
|
26
|
+
"read_config"]
|
|
23
27
|
|
|
24
28
|
CONFIG_KEY, SET_KEY = "$config", "$set"
|
|
25
29
|
_WEIGHTS = re.compile(r"\{weights(?:\.([A-Za-z_][A-Za-z0-9_-]*))?\}")
|
|
@@ -96,8 +100,25 @@ def _weights_of(part: str | None, weights: Any) -> str:
|
|
|
96
100
|
|
|
97
101
|
|
|
98
102
|
def build(cls: type, constructor: Any, *, base_dir: str, weights: Any) -> Any:
|
|
99
|
-
"""The instance `constructor` (a PolicyConstructor) makes of `cls`: `cls.<make>(*args, **kwargs)
|
|
103
|
+
"""The instance `constructor` (a PolicyConstructor) makes of `cls`: `cls.<make>(*args, **kwargs)`, the function
|
|
104
|
+
make="package.module:function" names called so, or `cls(...)`."""
|
|
100
105
|
args = expand(list(constructor.args), base_dir=base_dir, weights=weights)
|
|
101
106
|
kwargs = expand(dict(constructor.kwargs), base_dir=base_dir, weights=weights)
|
|
102
|
-
|
|
103
|
-
|
|
107
|
+
return maker(cls, constructor.make)(*args, **kwargs)
|
|
108
|
+
|
|
109
|
+
|
|
110
|
+
def maker(cls: type, make: str | None) -> Any:
|
|
111
|
+
"""What builds the instance: `cls` for no make=, its classmethod or staticmethod `make`, or - make written
|
|
112
|
+
"package.module:function" - that module's function, the module imported (as the policy's code imports it).
|
|
113
|
+
"""
|
|
114
|
+
if not make:
|
|
115
|
+
return cls
|
|
116
|
+
module, colon, name = make.rpartition(":")
|
|
117
|
+
if not colon:
|
|
118
|
+
return getattr(cls, make)
|
|
119
|
+
found = getattr(importlib.import_module(module), name, None)
|
|
120
|
+
if found is None:
|
|
121
|
+
raise ConstructError(f"make={make!r}: {module} has no function {name!r}")
|
|
122
|
+
if not callable(found):
|
|
123
|
+
raise ConstructError(f"make={make!r}: {module}.{name} is not callable ({type(found).__name__})")
|
|
124
|
+
return found
|
|
@@ -482,19 +482,26 @@ def check_weights_parts(weights_name: str | None, extra: dict) -> None:
|
|
|
482
482
|
raise ValueError(f"at most {MAX_WEIGHTS_PARTS} weights parts")
|
|
483
483
|
|
|
484
484
|
|
|
485
|
+
|
|
486
|
+
_MAKE = r"^(?:[A-Za-z_][A-Za-z0-9_]*(?:\.[A-Za-z_][A-Za-z0-9_]*)*:)?[A-Za-z_][A-Za-z0-9_]*$"
|
|
487
|
+
|
|
488
|
+
|
|
485
489
|
class PolicyConstructor(BaseModel):
|
|
486
490
|
"""How the worker builds a policy class that names its own methods (`@seq.policy(infer="get_action", ...)`): its
|
|
487
|
-
own constructor, or the factory it names (`make
|
|
488
|
-
|
|
489
|
-
|
|
490
|
-
|
|
491
|
-
|
|
491
|
+
own constructor, or the factory it names (`make`: a classmethod of it, e.g. LeRobot's `from_pretrained`, or a
|
|
492
|
+
module's function, "package.module:function", whose return is the instance its methods are called on), called
|
|
493
|
+
with `args` and `kwargs` where `@seq.load` would run - the replica's cold start, the weights on disk. The values are
|
|
494
|
+
the user's own, as JSON carries them, but for two forms the worker fills in first (sequence_base.construct.expand):
|
|
495
|
+
a config file's contents, `{"$config": <its path from the class's file>, "$set": {<keys laid over it>}}`, YAML or
|
|
496
|
+
JSON, and in any string `"{weights}"` (the weights directory) or `"{weights.<part>}"` (a named part's).
|
|
497
|
+
|
|
492
498
|
"""
|
|
493
499
|
|
|
494
500
|
model_config = ConfigDict(extra="forbid")
|
|
495
501
|
|
|
496
|
-
make: str | None = Field(default=None, pattern=
|
|
497
|
-
description="a classmethod or staticmethod that returns the instance
|
|
502
|
+
make: str | None = Field(default=None, max_length=512, pattern=_MAKE,
|
|
503
|
+
description="a classmethod or staticmethod that returns the instance, or a module's "
|
|
504
|
+
"function, 'package.module:function' (contract 0.5.27); None: the class")
|
|
498
505
|
args: list[Any] = Field(default_factory=list, description="positional arguments, in order")
|
|
499
506
|
kwargs: dict[str, Any] = Field(default_factory=dict, description="keyword arguments")
|
|
500
507
|
|
|
@@ -471,7 +471,7 @@
|
|
|
471
471
|
],
|
|
472
472
|
"title": "SeqFlags",
|
|
473
473
|
"type": "integer",
|
|
474
|
-
"x-contract-version": "0.5.
|
|
474
|
+
"x-contract-version": "0.5.27",
|
|
475
475
|
"x-flag-names": {
|
|
476
476
|
"CHECK": 65536,
|
|
477
477
|
"INFER": 16,
|
|
@@ -722,5 +722,5 @@
|
|
|
722
722
|
],
|
|
723
723
|
"title": "BenchmarkSpec",
|
|
724
724
|
"type": "object",
|
|
725
|
-
"x-contract-version": "0.5.
|
|
725
|
+
"x-contract-version": "0.5.27"
|
|
726
726
|
}
|
|
@@ -766,7 +766,7 @@
|
|
|
766
766
|
},
|
|
767
767
|
"PolicyConstructor": {
|
|
768
768
|
"additionalProperties": false,
|
|
769
|
-
"description": "How the worker builds a policy class that names its own methods (`@seq.policy(infer=\"get_action\", ...)`): its\nown constructor, or the factory it names (`make
|
|
769
|
+
"description": "How the worker builds a policy class that names its own methods (`@seq.policy(infer=\"get_action\", ...)`): its\nown constructor, or the factory it names (`make`: a classmethod of it, e.g. LeRobot's `from_pretrained`, or a\nmodule's function, \"package.module:function\", whose return is the instance its methods are called on), called\nwith `args` and `kwargs` where `@seq.load` would run - the replica's cold start, the weights on disk. The values are\nthe user's own, as JSON carries them, but for two forms the worker fills in first (sequence_base.construct.expand):\na config file's contents, `{\"$config\": <its path from the class's file>, \"$set\": {<keys laid over it>}}`, YAML or\nJSON, and in any string `\"{weights}\"` (the weights directory) or `\"{weights.<part>}\"` (a named part's).",
|
|
770
770
|
"properties": {
|
|
771
771
|
"args": {
|
|
772
772
|
"description": "positional arguments, in order",
|
|
@@ -783,7 +783,8 @@
|
|
|
783
783
|
"make": {
|
|
784
784
|
"anyOf": [
|
|
785
785
|
{
|
|
786
|
-
"
|
|
786
|
+
"maxLength": 512,
|
|
787
|
+
"pattern": "^(?:[A-Za-z_][A-Za-z0-9_]*(?:\\.[A-Za-z_][A-Za-z0-9_]*)*:)?[A-Za-z_][A-Za-z0-9_]*$",
|
|
787
788
|
"type": "string"
|
|
788
789
|
},
|
|
789
790
|
{
|
|
@@ -791,7 +792,7 @@
|
|
|
791
792
|
}
|
|
792
793
|
],
|
|
793
794
|
"default": null,
|
|
794
|
-
"description": "a classmethod or staticmethod that returns the instance; None: the class",
|
|
795
|
+
"description": "a classmethod or staticmethod that returns the instance, or a module's function, 'package.module:function' (contract 0.5.27); None: the class",
|
|
795
796
|
"title": "Make"
|
|
796
797
|
}
|
|
797
798
|
},
|
|
@@ -1172,7 +1173,7 @@
|
|
|
1172
1173
|
],
|
|
1173
1174
|
"title": "SeqFlags",
|
|
1174
1175
|
"type": "integer",
|
|
1175
|
-
"x-contract-version": "0.5.
|
|
1176
|
+
"x-contract-version": "0.5.27",
|
|
1176
1177
|
"x-flag-names": {
|
|
1177
1178
|
"CHECK": 65536,
|
|
1178
1179
|
"INFER": 16,
|
|
@@ -1363,5 +1364,5 @@
|
|
|
1363
1364
|
],
|
|
1364
1365
|
"title": "PolicyRegistration",
|
|
1365
1366
|
"type": "object",
|
|
1366
|
-
"x-contract-version": "0.5.
|
|
1367
|
+
"x-contract-version": "0.5.27"
|
|
1367
1368
|
}
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|