Sweep all 296 v2 modules: simplify verbose docstrings/comments and remove 376 dangling "(design_vN §X)" citations to the now-deleted design docs (v2/README.md is the source of truth). Comment/docstring-only — AST-verified code-identical; the CPU suite holds at 237 passed / 5 pre-existing. Also applies yapf + ruff --fix auto-fixes (import ordering, forward-ref annotation de-quoting under `from __future__ import annotations`; behavior- neutral, suite-confirmed) and adds the legitimate domain terms mot/clen/te to the codespell ignore-list. Remaining ruff (24) + mypy (68 no-untyped-call) findings are pre-existing v2 debt, untouched here.
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
"""TaskType — declared, never inferred.
|
|
|
|
The task is part of the typed request and the program branches on it. Heuristics
|
|
may only suggest a default at the product boundary; the runtime never infers it.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from enum import Enum
|
|
|
|
|
|
class TaskType(str, Enum):
|
|
# --- wired in phase 1 (Wan2.1 / LTX2) ---
|
|
T2V = "t2v" # text -> video
|
|
I2V = "i2v" # image -> video
|
|
TI2V = "ti2v" # text+image -> video
|
|
V2V = "v2v" # video -> video
|
|
T2I = "t2i" # text -> image (batchable diffusion)
|
|
|
|
# --- declared now for omni-readiness; implemented in phase 2 ---
|
|
T2A = "t2a" # text -> audio (diffusion)
|
|
T2VS = "t2vs" # text -> video+sound (joint denoise)
|
|
V2W = "v2w" # video -> world rollout (chunked, causal)
|
|
A2W = "a2w" # action -> world rollout
|
|
REASON = "reason" # AR text reasoning (Cosmos3 reasoner / omni thinker)
|
|
|
|
@property
|
|
def is_world_rollout(self) -> bool:
|
|
return self in (TaskType.V2W, TaskType.A2W)
|
|
|
|
@property
|
|
def is_autoregressive(self) -> bool:
|
|
return self in (TaskType.REASON, )
|