Full tests

[robsdavis]

Description

Fix timeout during full tests.

Affected Dependencies

None

How has this been tested?

• Describe the tests that you ran to verify your changes.
• Provide instructions so we can reproduce.
• List any relevant details for your test configuration.

Checklist

• [x] I have followed the Contribution Guidelines and Code of Conduct
• [x] I have commented my code following the van der Schaar Lab Styleguide
• [x] I have labelled this PR with the relevant Type labels
• [x] My changes are covered by tests