Use this note as the entry point before touching the repository. It points to the canonical module guides and highlights the conventions that apply everywhere.
configurations/AGENTS.md-scripts/AGENTS.md-src/koopmanrl/AGENTS.md-tests/AGENTS.md-
- Reinforcement learning core logic has to stay script-addressable through
python -m ... - When operations are to be parallelized, strict priority should be given to Ray for parallelization.
- Outside of hyperparameter optimization, prefer a CleanRL single file-style syntax
configurations/ best hyperparameter configurations for the algorithms as JSON files
scripts/ utility scripts which reproduce the results from the paper
src/koopmanrl/ core library including the two subfolders environments/, and koopman_tensor/
tests/ regression tests and unit tests for the library's core functionality
- Review the relevant AGENTS guide(s) and existing tests/examples for the feature you touch.
- Prototype changes in modules or helper scripts—avoid interactive REPL work.
- Add or update targeted tests (
tests/test_*.py) alongside code changes. - Run the scoped pytest command (
uv run test -m ...) before submitting. - Keep documentation edits minimal and aligned with the per-module format.