Skip to content

Main test

test_jax_can_use_the_GPU #

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_torch_can_use_the_GPU()

Test that torch can use the GPU if it we have one.

test_setting_just_algorithm_isnt_enough #

test_setting_just_algorithm_isnt_enough(
    experiment_dictconfig: DictConfig,
) -> None

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 #

test_run_auto_schema_via_cli_without_errors()

Checks that the command completes without errors.