diff --git a/.github/workflows/integration.yml b/.github/workflows/integration.yml index 402415c02a..8ed9790dca 100644 --- a/.github/workflows/integration.yml +++ b/.github/workflows/integration.yml @@ -3,14 +3,11 @@ name: integration on: repository_dispatch: type: [integration-test-command] - push: - branches: - - add-spatial-resample jobs: integration-py3: container: - image: nvcr.io/nvidia/pytorch:20.12-py3 # CUDA 11.1 + image: nvcr.io/nvidia/pytorch:21.12-py3 # CUDA 11.5 options: --gpus all runs-on: [self-hosted, linux, x64, common] steps: @@ -37,7 +34,7 @@ jobs: which python python -m pip install --upgrade pip wheel python -m pip uninstall -y torch torchvision - python -m pip install torch==1.11+cu111 torchvision==0.12+cu111 -f https://download.pytorch.org/whl/torch_stable.html + python -m pip install torch==1.11.0+cu115 torchvision==0.12.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html python -m pip install -r requirements-dev.txt - name: Run integration tests run: | diff --git a/tests/min_tests.py b/tests/min_tests.py index bb47403090..7d3b35be47 100644 --- a/tests/min_tests.py +++ b/tests/min_tests.py @@ -95,6 +95,7 @@ def run_testsuit(): "test_integration_unet_2d", "test_integration_workflows", "test_integration_workflows_gan", + "test_integration_bundle_run", "test_invertd", "test_iterable_dataset", "test_keep_largest_connected_component", @@ -159,7 +160,6 @@ def run_testsuit(): "test_zoomd", "test_prepare_batch_default_dist", "test_parallel_execution_dist", - "test_bundle_run", "test_bundle_verify_metadata", ] assert sorted(exclude_cases) == sorted(set(exclude_cases)), f"Duplicated items in {exclude_cases}" diff --git a/tests/test_bundle_run.py b/tests/test_integration_bundle_run.py similarity index 100% rename from tests/test_bundle_run.py rename to tests/test_integration_bundle_run.py diff --git a/tests/testing_data/inference.json b/tests/testing_data/inference.json index 6cc6de88ef..ce229023fb 100644 --- a/tests/testing_data/inference.json +++ b/tests/testing_data/inference.json @@ -6,11 +6,11 @@ "in_channels": 1, "out_channels": 2, "channels": [ - 16, - 32, - 64, - 128, - 256 + 2, + 2, + 4, + 8, + 4 ], "strides": [ 2, @@ -63,12 +63,12 @@ "inferer": { "_target_": "SlidingWindowInferer", "roi_size": [ - 96, - 96, - 96 + 64, + 64, + 32 ], "sw_batch_size": 4, - "overlap": 0.5 + "overlap": 0.25 }, "postprocessing": { "_target_": "Compose", diff --git a/tests/testing_data/inference.yaml b/tests/testing_data/inference.yaml index eb2870ee03..0a3383adbb 100644 --- a/tests/testing_data/inference.yaml +++ b/tests/testing_data/inference.yaml @@ -6,11 +6,11 @@ network_def: in_channels: 1 out_channels: 2 channels: - - 16 - - 32 - - 64 - - 128 - - 256 + - 2 + - 2 + - 4 + - 8 + - 4 strides: - 2 - 2 @@ -46,11 +46,11 @@ dataloader: inferer: _target_: SlidingWindowInferer roi_size: - - 96 - - 96 - - 96 + - 64 + - 64 + - 32 sw_batch_size: 4 - overlap: 0.5 + overlap: 0.25 postprocessing: _target_: Compose transforms: