Hongsheng Liu
339761045f
Close publication leases inside Host Weight Runtime, return only the publication report, and cover the new terminal mappings. Rename the runtime test module to avoid the pytest basename collision reported in #6436. Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>
144 lines
3.4 KiB
Python
144 lines
3.4 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
"""Neutral contracts for immutable runtime-ready host weight artifacts."""
|
|
|
|
from .config import (
|
|
CapacityPolicy,
|
|
HostWeightRuntimeConfig,
|
|
IntegrityPolicy,
|
|
ProductionPolicy,
|
|
RemoteImportPolicy,
|
|
RemoteOnMiss,
|
|
RuntimeMode,
|
|
StorageClass,
|
|
StorageDomainPolicy,
|
|
StorageScope,
|
|
ValidationLevel,
|
|
WaitPolicy,
|
|
)
|
|
from .errors import FailureCode, HostWeightError, HostWeightFailure, ResolutionAction, ResolutionStage
|
|
from .identity import (
|
|
AdaptationIdentity,
|
|
CanonicalJson,
|
|
ComponentIdentity,
|
|
IdentityValidationError,
|
|
ProducerIdentity,
|
|
RuntimeWeightLayout,
|
|
WeightArtifactIdentity,
|
|
WeightRepresentation,
|
|
WeightSourceIdentity,
|
|
)
|
|
from .inspection import (
|
|
ArtifactInspection,
|
|
ArtifactInventoryEntry,
|
|
ArtifactInventoryState,
|
|
DomainInspection,
|
|
HostWeightInspector,
|
|
)
|
|
from .lease import HostWeightLease, LeaseProvenance, MappedHostRegion
|
|
from .manifest import (
|
|
FileManifestEntry,
|
|
ManifestValidationError,
|
|
PublicationMarker,
|
|
TensorKind,
|
|
TensorManifestEntry,
|
|
WeightManifest,
|
|
)
|
|
from .outcomes import (
|
|
AttemptResult,
|
|
HostWeightResolution,
|
|
PostLoadPublicationOutcome,
|
|
PostLoadPublicationReport,
|
|
ResolutionAttempt,
|
|
ResolutionOutcome,
|
|
ResolutionReport,
|
|
StoreResult,
|
|
StoreStatus,
|
|
)
|
|
from .protocols import (
|
|
ArtifactWriter,
|
|
BuildRequest,
|
|
CoordinationScope,
|
|
HostWeightStore,
|
|
LookupPhase,
|
|
ProductionMetadata,
|
|
ProductionSourceMode,
|
|
RemoteArtifactDescriptor,
|
|
RemoteArtifactProvider,
|
|
TensorFileWriter,
|
|
TensorWriteSpec,
|
|
WeightProducer,
|
|
WeightProductionSpec,
|
|
WeightRestorePlan,
|
|
WeightRestorer,
|
|
)
|
|
from .runtime import HostWeightRuntime
|
|
|
|
__all__ = [
|
|
"AdaptationIdentity",
|
|
"ArtifactInspection",
|
|
"ArtifactInventoryEntry",
|
|
"ArtifactInventoryState",
|
|
"ArtifactWriter",
|
|
"AttemptResult",
|
|
"BuildRequest",
|
|
"CanonicalJson",
|
|
"CapacityPolicy",
|
|
"ComponentIdentity",
|
|
"CoordinationScope",
|
|
"DomainInspection",
|
|
"FailureCode",
|
|
"FileManifestEntry",
|
|
"HostWeightError",
|
|
"HostWeightFailure",
|
|
"HostWeightLease",
|
|
"HostWeightResolution",
|
|
"HostWeightInspector",
|
|
"HostWeightRuntime",
|
|
"HostWeightRuntimeConfig",
|
|
"HostWeightStore",
|
|
"IdentityValidationError",
|
|
"IntegrityPolicy",
|
|
"LeaseProvenance",
|
|
"LookupPhase",
|
|
"ManifestValidationError",
|
|
"MappedHostRegion",
|
|
"ProducerIdentity",
|
|
"PostLoadPublicationOutcome",
|
|
"PostLoadPublicationReport",
|
|
"ProductionMetadata",
|
|
"ProductionPolicy",
|
|
"ProductionSourceMode",
|
|
"PublicationMarker",
|
|
"RemoteArtifactDescriptor",
|
|
"RemoteArtifactProvider",
|
|
"RemoteImportPolicy",
|
|
"RemoteOnMiss",
|
|
"ResolutionAction",
|
|
"ResolutionAttempt",
|
|
"ResolutionOutcome",
|
|
"ResolutionReport",
|
|
"ResolutionStage",
|
|
"RuntimeMode",
|
|
"RuntimeWeightLayout",
|
|
"StorageClass",
|
|
"StorageDomainPolicy",
|
|
"StorageScope",
|
|
"StoreResult",
|
|
"StoreStatus",
|
|
"TensorFileWriter",
|
|
"TensorKind",
|
|
"TensorManifestEntry",
|
|
"TensorWriteSpec",
|
|
"ValidationLevel",
|
|
"WaitPolicy",
|
|
"WeightArtifactIdentity",
|
|
"WeightManifest",
|
|
"WeightProducer",
|
|
"WeightProductionSpec",
|
|
"WeightRepresentation",
|
|
"WeightRestorePlan",
|
|
"WeightRestorer",
|
|
"WeightSourceIdentity",
|
|
]
|