Main test
test_jax_can_use_the_GPU #
Test that Jax can use the GPU if it we have one.
test_torch_can_use_the_GPU #
Test that torch can use the GPU if it we have one.
test_setting_just_algorithm_isnt_enough #
Test to check that the datamodule is required (even when just the example algorithm is set).
TODO: We could probably move the datamodule
config under algorithm/datamodule
. Maybe that
would be better?
test_example_experiment_defaults #
test_example_experiment_defaults(
experiment_config: Config,
) -> None
Test to check that the datamodule is required (even when just an algorithm is set?!).
test_run_auto_schema_via_cli_without_errors #
Checks that the command completes without errors.