Skip to content

Commit

Permalink
Merge pull request #2364: Improved container priority resolution
Browse files Browse the repository at this point in the history
  • Loading branch information
MatthiasZepper authored Jul 17, 2023
2 parents f685aee + e459c66 commit 0fe54b0
Show file tree
Hide file tree
Showing 10 changed files with 265 additions and 90 deletions.
4 changes: 4 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,10 @@

### Template

### Download

- Improved container image resolution and prioritization of http downloads over Docker URIs ([#2364](https://github.com/nf-core/tools/pull/2364)).

### Linting

### Modules
Expand Down
206 changes: 120 additions & 86 deletions nf_core/download.py

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
process MOCK {
label 'process_fake'

conda (params.enable_conda ? "bioconda::singlequay=1.9" : null)
container "quay.io/biocontainers/singlequay:1.9--pyh9f0ad1d_0"

// truncated
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
process MOCK {
label 'process_fake'

conda "bioconda::dsltwoapptainervarone=1.1.0"
container "${ (workflow.containerEngine == 'singularity' || workflow.containerEngine == 'apptainer') && !task.ext.singularity_pull_docker_container ?
'https://depot.galaxyproject.org/singularity/dsltwoapptainervarone:1.1.0--py38h7be5676_2':
'biocontainers/dsltwoapptainervarone:1.1.0--py38h7be5676_2' }"

// truncated
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
process MOCK {
label 'process_fake'

conda "bioconda::dsltwoapptainervartwo=1.1.0"
container "${ ['singularity', 'apptainer'].contains(workflow.containerEngine) && !task.ext.singularity_pull_docker_container ?
'https://depot.galaxyproject.org/singularity/dsltwoapptainervartwo:1.1.0--hdfd78af_0':
'biocontainers/dsltwoapptainervartwo:1.1.0--hdfd78af_0' }"

// truncated
}
10 changes: 10 additions & 0 deletions tests/data/mock_module_containers/modules/mock_dsl2_current.nf
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
process MOCK {
label 'process_fake'

conda "bioconda::dsltwocurrent=1.2.1"
container "${ workflow.containerEngine == 'singularity' && !task.ext.singularity_pull_docker_container ?
'https://depot.galaxyproject.org/singularity/dsltwocurrent:1.2.1--pyhdfd78af_0':
'biocontainers/dsltwocurrent:1.2.1--pyhdfd78af_0' }"

// truncated
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
process MOCK {
label 'process_fake'

conda "bioconda::dsltwocurrentinv=3.3.2"
container "${ !workflow.containerEngine == 'singularity' && task.ext.singularity_pull_docker_container ?
'biocontainers/dsltwocurrentinv:3.3.2--h1b792b2_1' :
'https://depot.galaxyproject.org/singularity/dsltwocurrentinv:3.3.2--h1b792b2_1' }"

// truncated
}
12 changes: 12 additions & 0 deletions tests/data/mock_module_containers/modules/mock_dsl2_old.nf
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
process MOCK {
label 'process_fake'

conda (params.enable_conda ? "bioconda::dsltwoold=0.23.0" : null)
if (workflow.containerEngine == 'singularity' && !params.singularity_pull_docker_container) {
container "https://depot.galaxyproject.org/singularity/dsltwoold:0.23.0--0"
} else {
container "quay.io/biocontainers/dsltwoold:0.23.0--0"
}

// truncated
}
19 changes: 19 additions & 0 deletions tests/data/mock_module_containers/modules/mock_dsl2_variable.nf
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
process STAR_ALIGN {
// from rnaseq 3.7
label 'process_fake'

conda (params.enable_conda ? conda_str : null)
container "${ workflow.containerEngine == 'singularity' && !task.ext.singularity_pull_docker_container ?
"https://depot.galaxyproject.org/singularity/${container_id}" :
"quay.io/biocontainers/${container_id}" }"


// Note: 2.7X indices incompatible with AWS iGenomes so use older STAR version
conda_str = "bioconda::star=2.7.10a bioconda::samtools=1.15.1 conda-forge::gawk=5.1.0"
container_id = 'mulled-v2-1fa26d1ce03c295fe2fdcf85831a92fbcbd7e8c2:afaaa4c6f5b308b4b6aa2dd8e99e1466b2a6b0cd-0'
if (is_aws_igenome) {
conda_str = "bioconda::star=2.6.1d bioconda::samtools=1.10 conda-forge::gawk=5.1.0"
container_id = 'mulled-v2-1fa26d1ce03c295fe2fdcf85831a92fbcbd7e8c2:59cdd445419f14abac76b31dd0d71217994cbcc9-0'
}

}
66 changes: 62 additions & 4 deletions tests/test_download.py
Original file line number Diff line number Diff line change
Expand Up @@ -174,10 +174,68 @@ def test__find_container_images_config_nextflow(self, tmp_path, mock_fetch_wf_co
assert "nfcore/methylseq:1.4" in download_obj.containers
assert "nfcore/sarek:dev" in download_obj.containers
assert "https://depot.galaxyproject.org/singularity/r-shinyngs:1.7.1--r42hdfd78af_1" in download_obj.containers
# does not yet pick up nfcore/sarekvep:dev.${params.genome}, because successfully detecting "nfcore/sarek:dev"
# breaks the loop already. However, this loop-breaking is needed to stop iterating over DSL2 syntax if a
# direct download link has been found. Unless we employ a better deduplication, support for this kind of
# free-style if-else switches will sadly remain insufficient.
# does not yet pick up nfcore/sarekvep:dev.${params.genome}, because that is no valid URL or Docker URI.

#
# Test for 'find_container_images' in modules
#
@with_temporary_folder
@mock.patch("nf_core.utils.fetch_wf_config")
def test_find_container_images_modules(self, tmp_path, mock_fetch_wf_config):
download_obj = DownloadWorkflow(pipeline="dummy", outdir=tmp_path)
mock_fetch_wf_config.return_value = {}
download_obj.find_container_images(Path(__file__).resolve().parent / "data/mock_module_containers")

# mock_docker_single_quay_io.nf
assert "quay.io/biocontainers/singlequay:1.9--pyh9f0ad1d_0" in download_obj.containers

# mock_dsl2_apptainer_var1.nf (possible future convention?)
assert (
"https://depot.galaxyproject.org/singularity/dsltwoapptainervarone:1.1.0--py38h7be5676_2"
in download_obj.containers
)
assert "biocontainers/dsltwoapptainervarone:1.1.0--py38h7be5676_2" not in download_obj.containers

# mock_dsl2_apptainer_var2.nf (possible future convention?)
assert (
"https://depot.galaxyproject.org/singularity/dsltwoapptainervartwo:1.1.0--hdfd78af_0"
in download_obj.containers
)
assert "biocontainers/dsltwoapptainervartwo:1.1.0--hdfd78af_0" not in download_obj.containers

# mock_dsl2_current_inverted.nf (new implementation supports if the direct download URL is listed after Docker URI)
assert (
"https://depot.galaxyproject.org/singularity/dsltwocurrentinv:3.3.2--h1b792b2_1" in download_obj.containers
)
assert "biocontainers/dsltwocurrentinv:3.3.2--h1b792b2_1" not in download_obj.containers

# mock_dsl2_current.nf (main nf-core convention, should be the one in far the most modules)
assert (
"https://depot.galaxyproject.org/singularity/dsltwocurrent:1.2.1--pyhdfd78af_0" in download_obj.containers
)
assert "biocontainers/dsltwocurrent:1.2.1--pyhdfd78af_0" not in download_obj.containers

# mock_dsl2_old.nf (initial DSL2 convention)
assert "https://depot.galaxyproject.org/singularity/dsltwoold:0.23.0--0" in download_obj.containers
assert "quay.io/biocontainers/dsltwoold:0.23.0--0" not in download_obj.containers

# mock_dsl2_variable.nf (currently the edgiest edge case supported)
assert (
"https://depot.galaxyproject.org/singularity/mulled-v2-1fa26d1ce03c295fe2fdcf85831a92fbcbd7e8c2:59cdd445419f14abac76b31dd0d71217994cbcc9-0"
in download_obj.containers
)
assert (
"https://depot.galaxyproject.org/singularity/mulled-v2-1fa26d1ce03c295fe2fdcf85831a92fbcbd7e8c2:afaaa4c6f5b308b4b6aa2dd8e99e1466b2a6b0cd-0"
in download_obj.containers
)
assert (
"quay.io/biocontainers/mulled-v2-1fa26d1ce03c295fe2fdcf85831a92fbcbd7e8c2:59cdd445419f14abac76b31dd0d71217994cbcc9-0"
not in download_obj.containers
)
assert (
"quay.io/biocontainers/mulled-v2-1fa26d1ce03c295fe2fdcf85831a92fbcbd7e8c2:afaaa4c6f5b308b4b6aa2dd8e99e1466b2a6b0cd-0"
not in download_obj.containers
)

#
# Tests for 'singularity_pull_image'
Expand Down

0 comments on commit 0fe54b0

Please sign in to comment.