Nightly release

Merged PRs since last release:

🐛 Fixes

• tgolsson: use input_key in FeatureAgentProxy.input_names (#90)
• tgolsson: allow None in cycle to disable (#91)
• tgolsson: Add missing dataclass decorator on MetaData (#92)
• tgolsson: Improve error handling in memory (#98)
• tgolsson: fix tests (#103)
• spetravic: Missing return statement in AlphaLoss (#111)
• tgolsson: add atomicwrites dependency (#113)
• AliGhadirzadeh: Simona ali/tridays (#112)
• tgolsson: ensure memory exports are readable everywhere (#116)
• spetravic: Hyperbolic tangent instability (#118)
• tgolsson: ensure pending exports are processed per batch (#120)
• tgolsson: work around memory leak in torch.jit.trace (#126)
• tgolsson: detach images (#127)
• jaxs-ribs: Fix simple doc error (#131)
• jaxs-ribs: Add protocol=4 to cloudpickle.dump (#134)
• spetravic: Simona/fix logging (#144)
• tgolsson: update pdm-plugin-torch (#145)
• spetravic: Simona/onnx (#161)
• spetravic: Simona/add conditions (#162)

🧪 Tests

• tgolsson: fix tests (#103)

🧪 Development

• singhblom: Updated a few things missed in 3.8=>3.9 (#102)
• tgolsson: simplify lockfile error (#105)
• tgolsson: Improve CI error messages (#107)
• tgolsson: missing init file (#106)
• tgolsson: add final step to trigger other repo (#108)
• ghost: Ali/mbrl tests (#110)
• tgolsson: Add PDM 2.5.0 to CI and bump minimum version to 2.4.0 (#115)
• tgolsson: Fix branch filter for docs publish (#119)

Uncategorized:

• tgolsson: Add documentation about writing docs (#94)
• tgolsson: refactor LoggingCallback to LoggingMixin (#93)
• ghost: Gym is replaced by gymnasium (#95)
• singhblom: Ignore pypackages (#100)
• tgolsson: add LoggingProxyWrapper for the memory (#96)
• spetravic: Weights&biases logging integration (#104)
• ghost: Ali/mbrl lib (#101)
• AliGhadirzadeh: fixed the input_key error (#114)
• tgolsson: remove unused exitsignal (#124)
• tgolsson: Upgrade Memory Warmup callback to handle exports and early shutdowns (#125)
• jaxs-ribs: Add checkpointer_index constructor argument to Checkpointer (#128)
• tgolsson: fix version to work around urllib breakage (#121)
• jaxs-ribs: Add Storage Subdirectory (#130)
• tgolsson: split callbacks.py into a package (#109)
• spetravic: Add support for custom weight decay optimizer (#129)
• tgolsson: update torch to 1.12 (#133)
• tgolsson: Enable setting device for ONNX exporter (#135)
• jaxs-ribs: Add more logging to ckpt/onnx/memory (#136)
• tgolsson: always flush after writing tb events (#138)
• tgolsson: check for double-init episodes (#123)
• spetravic: Add logging (#140)
• tgolsson: fix docs build? (#141)
• spetravic: Agent proxy that supports multiple input keys (#143)
• spetravic: Add Conv1d encoder and update Conv2d encoder with option to flatten (#142)
• AliGhadirzadeh: updating the checkpointer and checkpoint loader (#146)
• klashenriksson: Fix alpha loss load_state_dict fn signature (#147)
• spetravic: Switch order of checks in log_windowed_scalar (#148)
• spetravic: Update windowed scalar (again) (#149)
• singhblom: Remove _inf_step from tb logs (#151)
• singhblom: Orthogonal init for conv as well (#152)
• spetravic: Change check for bias initialization (#156)
• tgolsson: add PDM 2.6 and 2.7 to CI image (#153)
• klashenriksson: add optional per-parameter logging for loss callbacks (#158)
• tgolsson: setup multi-python image (#160)
• spetravic: Simona/inits (#163)
• klashenriksson: Make MemoryProxyWrappers forward methods to wrapee if they do not exist on wrapper (#164)
• tgolsson: simplify agent proxy setup (#165)
• tgolsson: cherry-pick changes to reward-test-check (#166)
• tgolsson: Upgrade memory serialization workflows (#167)
• tgolsson: add OnnxExporter.add_metadata (#168)
• AliGhadirzadeh: Ali/genrl pr (#159)
• tgolsson: use same line-length as other repositories (#171)
• tgolsson: split state restoration from begin_training (#172)
• tgolsson: run flake8 on CI (#173)
• klashenriksson: Allow temp storage to grow to fit the largest requested data (#174)
• jaxs-ribs: Fix Lunar Lander Test (#177)
• jaxs-ribs: Implement DQN and CartPole Test (#175)
• tgolsson: log memory timers (#179)
• jaxs-ribs: Implement Coverage Based Sampling Strategy (#178)
• jaxs-ribs: Add "latest_checkpoint" path to state_dict (#169)
• jaxs-ribs: Add Documentation for Cartpole Training (#180)
• spetravic: Add schedules (#185)
• spetravic: Rename get_last_val (#187)
• spetravic: Update AlphaLoss to support entropy schedules (#186)
• jaxs-ribs: Fix wrong onnx.load_model arguments (#188)
• tgolsson: fix bug in logging timers (#189)
• tgolsson: add pdm 2.9 and 2.10 to CI image (#190)
• jaxs-ribs: Upgrade to Python 3.10 (#184)
• tgolsson: Upgrade to PDM 2.10 (#191)
• jaxs-ribs: Update Github Workflows for Python3.10 (#192)
• jaxs-ribs: Properly set workflow file (#193)
• klashenriksson: Make LoggingProxyWrapper report inf_step if requested (#194)
• AliGhadirzadeh: Ali/amp (#195)
• klashenriksson: Add a joint memory loader capable of sampling data from multiple memory loaders (#181)
• AliGhadirzadeh: adding JointMemoryLoader to the init (#196)
• klashenriksson: Add NextNElementWrapper virtual storage (#198)