Skip to content

Commit 2c36ed7

Browse files
author
Xiangyu Lu
committed
Remove eval CLI references
1 parent 003d66d commit 2c36ed7

5 files changed

Lines changed: 4 additions & 15 deletions

File tree

.agents/skills/cosmos3-codebase-nav/SKILL.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,6 @@ Users can also supply a custom defaults file per-request via the `defaults_file`
9595
| ---------------------------------- | -------------------------------------------------------------------------------------------- |
9696
| Batch inference | `python -m cosmos_framework.scripts.inference` |
9797
| Training | `python -m cosmos_framework.scripts.train --sft-toml=examples/toml/sft_config/<recipe>.toml` |
98-
| Action evaluation | `python -m cosmos_framework.scripts.eval` |
9998
| Online serving (Ray) | `python -m cosmos_framework.inference.ray.serve` |
10099
| Submit to Ray server | `python -m cosmos_framework.inference.ray.submit` |
101100
| Gradio UI | `python -m cosmos_framework.inference.ray.gradio` |

AGENTS.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ Read this file first — it is the canonical map for navigating the Cosmos repos
77
- **Training infrastructure** — top-level subpackages under `cosmos_framework/` (data, model, trainer, callbacks, checkpoint, …).
88
- **Inference infrastructure**`cosmos_framework/inference/` (Diffusers / Transformers / vLLM-friendly inference core, online serving via Ray + Gradio).
99
- **Backend packages**`packages/{diffusers,transformers,vllm}-cosmos3/` provide library-style shims that load Cosmos3 checkpoints into the respective ecosystems.
10-
- **Entry-point scripts**`cosmos_framework/scripts/` (`train.py`, `inference.py`, `eval.py`, `export_model.py`, …) invoked as `python -m cosmos_framework.scripts.<name>`. Primary training entry point: `cosmos_framework.scripts.train` driven by a structured, pydantic-validated TOML interface (`--sft-toml=<recipe-toml>`); the schema lives at [`cosmos_framework/configs/toml_config/sft_config.py`](./cosmos_framework/configs/toml_config/sft_config.py) and the canonical recipe pattern is documented in [`examples/README.md`](./examples/README.md).
10+
- **Entry-point scripts**`cosmos_framework/scripts/` (`train.py`, `inference.py`, `export_model.py`, …) invoked as `python -m cosmos_framework.scripts.<name>`. Primary training entry point: `cosmos_framework.scripts.train` driven by a structured, pydantic-validated TOML interface (`--sft-toml=<recipe-toml>`); the schema lives at [`cosmos_framework/configs/toml_config/sft_config.py`](./cosmos_framework/configs/toml_config/sft_config.py) and the canonical recipe pattern is documented in [`examples/README.md`](./examples/README.md).
1111

1212
> All paths below are relative to the repository root (the directory containing `pyproject.toml`, the `cosmos_framework/` Python package, and `packages/`).
1313

cosmos_framework/inference/_test.py

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
from typing_extensions import Self
1414

1515
from cosmos_framework.data.vfm.utils import VIDEO_RES_SIZE_INFO
16-
from cosmos_framework.inference.metrics import compute_action_mse, compute_psnr
1716
from cosmos_framework.inference.args import (
1817
IMAGE_ONLY_RESOLUTIONS,
1918
AspectRatio,
@@ -27,6 +26,7 @@
2726
from cosmos_framework.inference.common.checkpoints import CHECKPOINTS, DATASETS, DatasetConfig
2827
from cosmos_framework.inference.fixtures.args import MAX_GPUS
2928
from cosmos_framework.inference.fixtures.script import INPUT_DIR, ScriptConfig, ScriptRunner, script_test
29+
from cosmos_framework.inference.metrics import compute_action_mse, compute_psnr
3030
from cosmos_framework.utils.checkpoint_db import CheckpointConfig
3131

3232
_CURRENT_DIR = Path(__file__).parent.absolute()
@@ -394,15 +394,6 @@ def __call__(self, runner: ScriptRunner, cfg: ScriptConfig) -> dict[str, str]:
394394
marks=marks,
395395
get_env=get_env,
396396
),
397-
ScriptConfig(
398-
name=f"eval_{name}",
399-
script=_TEST_DIR / "eval.sh",
400-
levels=(0, 2),
401-
gpus=(1, MAX_GPUS, MAX_GPUS),
402-
marks=marks,
403-
get_env=get_env,
404-
after_script=_action_after_script,
405-
),
406397
]
407398

408399

docs/code_structure.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ Cosmos/
3737
├── cosmos_framework/ # Main package (training infra + inference subpackage)
3838
│ ├── inference/ # Inference subpackage (model, args, defaults, Ray serving, common helpers, SFT experiment configs)
3939
│ └── ... # Training-infra subpackages: data, model, trainer, callbacks, checkpoint, …
40-
│ └── scripts/ # CLI entry-point scripts: train.py, _train.py, inference.py, eval.py, export_model.py, …
40+
│ └── scripts/ # CLI entry-point scripts: train.py, _train.py, inference.py, export_model.py, …
4141
├── packages/ # Backend shim packages: diffusers-cosmos3, transformers-cosmos3, vllm-cosmos3
4242
├── docs/ # User documentation (you are here)
4343
├── docker/ # Dockerfiles for reproducible environments
@@ -196,7 +196,7 @@ Add new worker types as sibling subpackages — each owns its own startup, messa
196196
- `tests/` — pytest tests, mirroring the `cosmos_framework/` package layout.
197197
- `examples/` — runnable end-to-end examples; see `examples/README.md`.
198198
- `docker/` — Dockerfiles and image build helpers; see `docker/README.md`.
199-
- `cosmos_framework/scripts/` — CLI entry-point scripts (`train.py`, `inference.py`, `eval.py`, `export_model.py`, …); invoke as `python -m cosmos_framework.scripts.<name>`. Primary training entry point: `cosmos_framework.scripts.train` driven by a structured, pydantic-validated TOML interface (`--sft-toml=<recipe-toml>`); recipe TOMLs live under [`examples/toml/sft_config/`](../examples/toml/sft_config/) and the schema is defined in [`cosmos_framework/configs/toml_config/sft_config.py`](../cosmos_framework/configs/toml_config/sft_config.py) — see [`examples/README.md`](../examples/README.md) and [`docs/training.md`](./training.md).
199+
- `cosmos_framework/scripts/` — CLI entry-point scripts (`train.py`, `inference.py`, `export_model.py`, …); invoke as `python -m cosmos_framework.scripts.<name>`. Primary training entry point: `cosmos_framework.scripts.train` driven by a structured, pydantic-validated TOML interface (`--sft-toml=<recipe-toml>`); recipe TOMLs live under [`examples/toml/sft_config/`](../examples/toml/sft_config/) and the schema is defined in [`cosmos_framework/configs/toml_config/sft_config.py`](../cosmos_framework/configs/toml_config/sft_config.py) — see [`examples/README.md`](../examples/README.md) and [`docs/training.md`](./training.md).
200200
- `packages/` — library-style backend shims: `packages/{diffusers,transformers,vllm}-cosmos3/`, each installable independently.
201201

202202
## Where to Add New Code

docs/setup.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,6 @@ If there is no attention backend wheel for your torch/cuda versions, you can bui
216216
Optional package extras:
217217

218218
- `train`: Training infrastructure (FSDP, parallelism, checkpointing, datasets)
219-
- `eval`: Evaluation harnesses for trained checkpoints
220219

221220
#### CUDA Variants
222221

0 commit comments

Comments
 (0)