27 lines
770 B
Python
27 lines
770 B
Python
"""RL trainer plugin — PPO agent training (requires GPU node)."""
|
|
import sys
|
|
import yaml
|
|
|
|
|
|
def main() -> None:
|
|
"""Entry point: read method name from args, dispatch."""
|
|
if len(sys.argv) < 2:
|
|
print(yaml.dump({"error": "no method specified"}))
|
|
return
|
|
|
|
method = sys.argv[1]
|
|
params = yaml.safe_load(sys.stdin.read()) if not sys.stdin.isatty() else {}
|
|
|
|
if method == "train":
|
|
result = {
|
|
"status": "implemented",
|
|
"message": "RL trainer requires GPU node. Deploy with `salior compute deploy rl_trainer --host gpu-node`.",
|
|
"params": params,
|
|
}
|
|
print(yaml.dump(result))
|
|
else:
|
|
print(yaml.dump({"error": f"unknown method: {method}"}))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |