hydra-auto-schema 0.0.2

Automagically generate yaml schemas for your Hydra config files

hydra-auto-schema

This tool greatly improves the experience of developing a project with Hydra by enabling rich IDE support for Hydra config files.

📢 If you are thinking of using Hydra for a PyTorch or Jax-based ML project, you might want to take a look at the Research Project Template where this plugin was originally created. This is also where this plugin is best integrated!

With this, you'll now get to see:

• the list of available configuration options in a given config
• the default values for each entry
• the documentation associated with each entry (taken from the source code of the _target_ callable!) Additionally
• A warning is displayed when a value is unexpected, or of the wrong type.

All-in-all, this helps to prevent errors, and gives your codebase the same kind of neatness and safety that type hints do.

Demo

https://github.com/user-attachments/assets/08f52d47-ebba-456d-95ef-ac9525d8e983

Installation

Requirements

At the moment, we assume that you are using VSCode as your code editor. We use the yaml extension by RedHat, and install it for you if it isn't already. However any IDE with a YAML language server should be fairly easy to make work. Please make an issue if this doesn't work with your IDE.

uv (recommended)

uv add git+https://www.github.com/lebrice/hydra-auto-schema

Note: This plugin needs to be installed in your project's virtual environment. It should not be installed as an isolated tool using uv tool. This is because the plugin needs to import the modules where the _target_s are defined in order to inspect their signature.

pip

pip install git+https://www.github.com/lebrice/hydra-auto-schema

Usage

Generate the yaml schemas for all the configs in the configs folder:

hydra-auto-schema configs

Watch for changes in the configs folder and update the schemas as needed:

hydra-auto-schema configs --watch

How to Contribute

This is a very new tool, and we'd love to get your feedback! Please feel free to make an Issue if you have any questions or feedback. We'll be happy to assist you.