cli

cli#

Functions#

_render_single_file_template

Returns a single-file algorithm scaffold containing:

_to_snake

Convert a class or algorithm name to snake_case.

generate_single_file_algorithm

main

prt_rl.cli._render_single_file_template(algo_pascal: str) → str[source]#
Returns a single-file algorithm scaffold containing:

  • Config dataclass

  • HeadSpec + PolicySpec dataclasses

  • PolicyModule implementation

  • PolicyFactory implementation

  • Agent implementation with save/load skeleton

Minimal dependencies: dataclasses, json, pathlib, torch, numpy (optional) and your prt_rl primitives (BaseAgent, EnvParams, PolicyModule, heads, etc.)

prt_rl.cli._to_snake(name: str) → str[source]#

Convert a class or algorithm name to snake_case.

Rules:

  • Preserve acronyms with digits: A3C -> a3c, TD3 -> td3

  • Preserve all-caps acronyms: PPO -> ppo, SAC -> sac

  • Convert CamelCase: SoftActorCritic -> soft_actor_critic

prt_rl.cli.generate_single_file_algorithm(agent_path: str, *, repo_root: Path | None = None, force: bool = False) → Path[source]#
prt_rl.cli.main(argv: List[str] | None = None) → None[source]#
Contents