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_run_auto_schema_via_cli_without_errors #
Checks that the command completes without errors.