Add RoseTTAFold-All-Atom

.github/CONTRIBUTING.md

@@ -29,7 +29,7 @@ If you're not used to this workflow with git, you can start with some [docs from
 You have the option to test your changes locally by running the pipeline. For receiving warnings about process selectors and other `debug` information, it is recommended to use the debug profile. Execute all the tests with the following command:
 
 ```bash
-nextflow run . --profile debug,test,docker --outdir <OUTDIR>
+nextflow run . -profile debug,test,docker --outdir <OUTDIR>
 ```
 
 When you create a pull request with changes, [GitHub Actions](https://github.com/features/actions) will run automatic tests.
@@ -78,8 +78,8 @@ If you wish to contribute a new step, please use the following coding standards:
 5. Add any new parameters to `nextflow_schema.json` with help text (via the `nf-core pipelines schema build` tool).
 6. Add sanity checks and validation for all relevant parameters.
 7. Perform local tests to validate that the new code works as expected.
-8. If applicable, add a new test command in `.github/workflow/ci.yml`.
-9. Update MultiQC config `assets/multiqc_config.yml` so relevant suffixes, file name clean up and module plots are in the appropriate order. If applicable, add a [MultiQC](https://https://multiqc.info/) module.
+8. If applicable, add a new test command in `.github/workflows/ci.yml`.
+9. Update MultiQC config `assets/multiqc_config.yml` so relevant suffixes, file name clean up and module plots are in the appropriate order. If applicable, add a [MultiQC](https://multiqc.info/) module.
 10. Add a description of the output files and if relevant any appropriate images from the MultiQC report to `docs/output.md`.
 
 ### Default values

.github/workflows/ci.yml

@@ -44,6 +44,7 @@ jobs:
           - "test_colabfold_download"
           - "test_esmfold"
           - "test_split_fasta"
+          - "test_rosettafold_all_atom"
         isMaster:
           - ${{ github.base_ref == 'master' }}
         # Exclude conda and singularity on dev

CHANGELOG.md

@@ -13,8 +13,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - [[#180](https://github.com/nf-core/proteinfold/issues/180)] - Implement Fooldseek.
 - [[#188](https://github.com/nf-core/proteinfold/issues/188)] - Fix colabfold image to run in gpus.
 - [[PR ##205](https://github.com/nf-core/proteinfold/pull/205)] - Change input schema from `sequence,fasta` to `id,fasta`.
-- [[PR #210](https://github.com/nf-core/proteinfold/pull/210)] - Moving post-processing logic to a subworkflow, change wave images pointing to oras to point to https and refactor module to match nf-core folder structure.
-- [[#214](https://github.com/nf-core/proteinfold/issues/214)] - Fix colabfold image to run in cpus after [#188](https://github.com/nf-core/proteinfold/issues/188) fix.
+- [[PR #210](https://github.com/nf-core/proteinfold/pull/210)]- Moving post-processing logic to a subworkflow, change wave images pointing to oras to point to https and refactor module to match nf-core folder structure.
+- [[#214](https://github.com/nf-core/proteinfold/issues/214)]- Fix colabfold image to run in cpus after [#188](https://github.com/nf-core/proteinfold/issues/188) fix.
+- [[PR ##220](https://github.com/nf-core/proteinfold/pull/220)] - Add RoseTTAFold-All-Atom module.
 - [[#235](https://github.com/nf-core/proteinfold/issues/235)] - Update samplesheet to new version (switch from `sequence` column to `id`).
 
 ## [[1.1.1](https://github.com/nf-core/proteinfold/releases/tag/1.1.1)] - 2025-07-30
@@ -106,6 +107,7 @@ Thank you to everyone else that has contributed by reporting bugs, enhancements
 |                       | `--esm2_t36_3B_UR50D_contact_regression` |
 |                       | `--esmfold_params_path`                  |
 |                       | `--skip_multiqc`                         |
+|                       | `--rosettafold_all_atom_db`              |
 
 > **NB:** Parameter has been **updated** if both old and new parameter information is present.
 > **NB:** Parameter has been **added** if just the new parameter information is present.

README.md

@@ -39,6 +39,8 @@ On release, automated continuous integration tests run the pipeline on a full-si
 
    v. [ESMFold](https://github.com/facebookresearch/esm) - Regular ESM
 
+   vi. [RoseTTAFold-All-Atom](https://github.com/baker-laboratory/RoseTTAFold-All-Atom/) - Regular RFAA
+
 ## Usage
 
 > [!NOTE]
@@ -53,7 +55,7 @@ nextflow run nf-core/proteinfold \
    --outdir <OUTDIR>
 ```
 
-The pipeline takes care of downloading the databases and parameters required by AlphaFold2, Colabfold or ESMFold. In case you have already downloaded the required files, you can skip this step by providing the path to the databases using the corresponding parameter [`--alphafold2_db`], [`--colabfold_db`] or [`--esmfold_db`]. Please refer to the [usage documentation](https://nf-co.re/proteinfold/usage) to check the directory structure you need to provide for each of the databases.
+The pipeline takes care of downloading the databases and parameters required by AlphaFold2, Colabfold or ESMFold. In case you have already downloaded the required files, you can skip this step by providing the path to the databases using the corresponding parameter [`--alphafold2_db`], [`--colabfold_db`], [`--esmfold_db`] or ['--rosettafold_all_atom_db']. Please refer to the [usage documentation](https://nf-co.re/proteinfold/usage) to check the directory structure you need to provide for each of the databases.
 
 - The typical command to run AlphaFold2 mode is shown below:
 
@@ -136,6 +138,18 @@ The pipeline takes care of downloading the databases and parameters required by
       -profile <docker/singularity/podman/shifter/charliecloud/conda/institute>
   ```
 
+- The rosettafold_all_atom mode can be run using the command below:
+
+  ```console
+  nextflow run nf-core/proteinfold \
+      --input samplesheet.csv \
+      --outdir <OUTDIR> \
+      --mode rosettafold_all_atom \
+      --rosettafold_all_atom_db <null (default) | PATH> \
+      --use_gpu <true/false> \
+      -profile <docker/singularity/podman/shifter/charliecloud/conda/institute>
+  ```
+
 > [!WARNING]
 > Please provide pipeline parameters via the CLI or Nextflow `-params-file` option. Custom config files including those provided by the `-c` Nextflow option can be used to provide any configuration _**except for parameters**_; see [docs](https://nf-co.re/docs/usage/getting_started/configuration#custom-configuration-files).
 

assets/schema_input.json

@@ -7,6 +7,12 @@
     "items": {
         "type": "object",
         "properties": {
+            "sequence": {
+                "type": "string",
+                "pattern": "^\\S+$",
+                "errorMessage": "Sequence name must be provided and cannot contain spaces",
+                "meta": ["sequence"]
+            },
             "id": {
                 "type": "string",
                 "pattern": "^\\S+$",
@@ -17,10 +23,11 @@
                 "type": "string",
                 "format": "file-path",
                 "exists": true,
-                "pattern": "^\\S+\\.fa(sta)?$",
-                "errorMessage": "Fasta file must be provided, cannot contain spaces and must have extension '.fa' or '.fasta'"
+                "pattern": "^\\S+\\.(fa(sta)?|yaml|yml|json)$",
+                "errorMessage": "Fasta, yaml or json file must be provided, cannot contain spaces and must have extension '.fa', '.fasta', '.yaml', '.yml', or '.json'"
             }
         },
-        "required": ["id", "fasta"]
+        "required": ["fasta"],
+        "anyOf": [{ "required": ["sequence"] }, { "required": ["id"] }]
     }
 }

bin/generate_report.py

@@ -307,6 +307,7 @@ def pdb_to_lddt(pdb_files, generate_tsv):
     "esmfold": "ESMFold",
     "alphafold2": "AlphaFold2",
     "colabfold": "ColabFold",
+    "rosettafold_all_atom": "Rosettafold_All_Atom",
 }
 
 parser = argparse.ArgumentParser()

conf/dbs.config

@@ -48,6 +48,18 @@ params {
         "alphafold2_ptm"         : "alphafold_params_2021-07-14"
     ]
 
+    // RoseTTAFold_All_Atom links
+    uniref30_rosettafold_all_atom_link  = 'http://wwwuser.gwdg.de/~compbiol/uniclust/2020_06/UniRef30_2020_06_hhsuite.tar.gz'
+    pdb100_rosettafold_all_atom_link    = 'https://files.ipd.uw.edu/pub/RoseTTAFold/pdb100_2021Mar03.tar.gz'
+    bfd_rosettafold_all_atom_link       = 'https://bfd.mmseqs.com/bfd_metaclust_clu_complete_id30_c90_final_seq.sorted_opt.tar.gz'
+    rfaa_paper_weights_link             = 'http://files.ipd.uw.edu/pub/RF-All-Atom/weights/RFAA_paper_weights.pt'
+
+    // RoseTTAFold_All_Atom paths
+    uniref30_rosettafold_all_atom_path  = "${params.rosettafold_all_atom_db}/uniref30/UniRef30_2020_06/*"
+    pdb100_rosettafold_all_atom_path    = "${params.rosettafold_all_atom_db}/pdb100_2021Mar03/*"
+    bfd_rosettafold_all_atom_path       = "${params.rosettafold_all_atom_db}/bfd/*"
+    rfaa_paper_weights_path             = "${params.rosettafold_all_atom_db}/RFAA_paper_weights.pt"
+
     // Esmfold links
     esmfold_3B_v1                        = 'https://dl.fbaipublicfiles.com/fair-esm/models/esmfold_3B_v1.pt'
     esm2_t36_3B_UR50D                    = 'https://dl.fbaipublicfiles.com/fair-esm/models/esm2_t36_3B_UR50D.pt'

conf/modules_rosettafold_all_atom.config

@@ -0,0 +1,22 @@
+/*
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+    Config file for defining DSL2 per module options and publishing paths
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+    Available keys to override module options:
+        ext.args   = Additional arguments appended to command in module.
+        ext.args2  = Second set of arguments appended to command in module (multi-tool modules).
+        ext.args3  = Third set of arguments appended to command in module (multi-tool modules).
+        ext.prefix = File name prefix for output files.
+----------------------------------------------------------------------------------------
+*/
+
+process {
+    withName: 'NFCORE_PROTEINFOLD:ROSETTAFOLD_ALL_ATOM:MULTIQC' {
+        publishDir = [
+            path: { "${params.outdir}/multiqc" },
+            mode: 'copy',
+            saveAs: { filename -> filename.equals('versions.yml') ? null : "rosettafold_all_atom_$filename" }
+        ]
+    }
+
+}

conf/test_rosettafold_all_atom.config

@@ -0,0 +1,37 @@
+/*
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+    Nextflow config file for running minimal tests
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+    Defines input files and everything required to run a fast and simple pipeline test.
+    Use as follows:
+        nextflow run nf-core/proteinfold -profile test_rosettafold_all_atom,<docker/singularity> --outdir <OUTDIR>
+----------------------------------------------------------------------------------------
+*/
+
+stubRun = true
+
+// Limit resources so that this can run on GitHub Actions
+process {
+    resourceLimits = [
+        cpus: 4,
+        memory: '15.GB',
+        time: '1.h'
+    ]
+}
+
+params {
+    config_profile_name        = 'Test profile'
+    config_profile_description = 'Minimal test dataset to check pipeline function'
+
+    // Input data to test rosettafold_all_atom
+    mode             = 'rosettafold_all_atom'
+    rosettafold_all_atom_db       = "${projectDir}/assets/dummy_db_dir"
+    input            = params.pipelines_testdata_base_path + 'proteinfold/testdata/samplesheet/v1.0/samplesheet.csv'
+}
+
+process {
+    withName: 'RUN_ROSETTAFOLD_ALL_ATOM' {
+        container = '/srv/scratch/sbf-pipelines/proteinfold/singularity/rosettafold_all_atom.sif'
+    }
+}
+

dockerfiles/Dockerfile_nfcore-proteinfold_rosettafold_all_atom

@@ -0,0 +1,35 @@
+FROM nvidia/cuda:12.6.0-cudnn-devel-ubuntu24.04
+
+LABEL Author="[email protected]" \
+    title="nfcore/proteinfold_rosettafold_all_atom" \
+    Version="0.9.0" \
+    description="Docker image containing all software requirements to run the RUN_ROSETTAFOLD_ALL_ATOM module using the nf-core/proteinfold pipeline"
+
+ENV PYTHONPATH="/app/RoseTTAFold-All-Atom" \
+    PATH="/conda/bin:/app/RoseTTAFold-All-Atom:$PATH" \
+    DGLBACKEND="pytorch" \
+    LD_LIBRARY_PATH="/conda/lib:/usr/local/cuda-12.6/lib64:$LD_LIBRARY_PATH"
+
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install --no-install-recommends -y wget git && \
+    wget -q -P /tmp "https://github.com/conda-forge/miniforge/releases/latest/download/Miniforge3-$(uname)-$(uname -m).sh" && \
+    bash /tmp/Miniforge3-$(uname)-$(uname -m).sh -b -p /conda && \
+    rm -rf /tmp/Miniforge3-$(uname)-$(uname -m).sh /var/lib/apt/lists/* && \
+    apt-get autoremove -y && apt-get clean -y
+
+RUN git clone --single-branch --depth 1 https://github.com/Australian-Structural-Biology-Computing/RoseTTAFold-All-Atom.git /app/RoseTTAFold-All-Atom && \
+    cd /app/RoseTTAFold-All-Atom && \
+    /conda/bin/mamba env create --file=environment.yaml && \
+    /conda/bin/mamba run -n RFAA bash -c \
+        "python /app/RoseTTAFold-All-Atom/rf2aa/SE3Transformer/setup.py install && \
+        bash /app/RoseTTAFold-All-Atom/install_dependencies.sh" && \
+    /conda/bin/mamba clean --all --force-pkgs-dirs -y
+
+RUN cd /app/RoseTTAFold-All-Atom && \
+    wget https://ftp.ncbi.nlm.nih.gov/blast/executables/legacy.NOTSUPPORTED/2.2.26/blast-2.2.26-x64-linux.tar.gz && \
+    mkdir -p blast-2.2.26 && \
+    tar -xf blast-2.2.26-x64-linux.tar.gz -C blast-2.2.26 && \
+    cp -r blast-2.2.26/blast-2.2.26/ blast-2.2.26_bk && \
+    rm -r blast-2.2.26 && \
+    mv blast-2.2.26_bk/ blast-2.2.26 && \
+    rm -rf /root/.cache *.tar.gz && \
+    apt-get autoremove -y && apt-get remove --purge -y wget git && apt-get clean -y

dockerfiles/rosettafold_all_atom.def

@@ -0,0 +1,41 @@
+Bootstrap: docker
+From: nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
+
+%labels
+    Author [email protected]
+    Version 0.2.5
+
+%post
+    apt update && DEBIAN_FRONTEND=noninteractive apt install --no-install-recommends -y wget git build-essential
+
+    wget https://github.com/conda-forge/miniforge/releases/latest/download/Miniforge3-Linux-x86_64.sh
+    bash Miniforge3-Linux-x86_64.sh -b -p /opt/miniforge
+    rm Miniforge3-Linux-x86_64.sh
+    export PATH="/opt/miniforge/bin:$PATH"
+
+    git clone --single-branch --depth 1 https://github.com/Australian-Structural-Biology-Computing/RoseTTAFold-All-Atom.git /app/RoseTTAFold-All-Atom
+    cd /app/RoseTTAFold-All-Atom
+    mamba env create --file=environment.yaml
+
+    mamba run -n RFAA \
+        'python rf2aa/SE3Transformer/setup.py install && \
+        bash install_dependencies.sh'
+
+    wget https://ftp.ncbi.nlm.nih.gov/blast/executables/legacy.NOTSUPPORTED/2.2.26/blast-2.2.26-x64-linux.tar.gz
+    mkdir -p blast-2.2.26
+    tar -xf blast-2.2.26-x64-linux.tar.gz -C blast-2.2.26
+    cp -r blast-2.2.26/blast-2.2.26/ blast-2.2.26_bk
+    rm -r blast-2.2.26
+    mv blast-2.2.26_bk/ blast-2.2.26
+
+    apt autoremove -y && apt remove --purge -y wget git build-essential && apt clean -y
+    rm -rf /var/lib/apt/lists/* /root/.cache *.tar.gz
+    mamba clean --all --force-pkgs-dirs -y
+
+%environment
+    export PYTHONPATH="/app/RoseTTAFold-All-Atom:$PYTHONPATH"
+    export PATH="/opt/miniforge/bin:/app/RoseTTAFold-All-Atom:$PATH"
+    export DGLBACKEND="pytorch"
+
+%runscript
+    mamba run --name RFAA python -m rf2aa.run_inference --config-name "$@"

docs/output.md

@@ -13,6 +13,7 @@ The pipeline is built using [Nextflow](https://www.nextflow.io/) and predicts pr
 - [AlphaFold2](https://github.com/deepmind/alphafold)
 - [ColabFold](https://github.com/sokrypton/ColabFold) - MMseqs2 (API server or local search) followed by ColabFold
 - [ESMFold](https://github.com/facebookresearch/esm)
+- [RoseTTAFold-All-Atom](https://github.com/baker-laboratory/RoseTTAFold-All-Atom/)
 
 See main [README.md](https://github.com/nf-core/proteinfold/blob/master/README.md) for a condensed overview of the steps in the pipeline, and the bioinformatics tools used at each step.
 
@@ -176,6 +177,19 @@ Below you can find an indicative example of the TSV file with the pLDDT scores p
 | 49                 | CB        | VAL          | 7                       | 52.74 |
 | 50                 | O         | VAL          | 7                       | 56.46 |
 
+### RoseTTAFold-All-Atom
+
+<details markdown="1">
+<summary>Output files</summary>
+
+- `run/`
+  - `<SEQUENCE NAME>_rosettafold_all_atom.pdb` that is the structure with the highest pLDDT score (ranked first)
+  - `<SEQUENCE NAME>_plddt_mqc.tsv` that presents the pLDDT scores per residue for the predicted model
+  - `<SEQUENCE NAME>_aux.pt` pytorch file with confidence metrics stored (can load with torch.load(file, map_location="cpu"))
+  - `<SEQUENCE NAME>/` that contains the computed MSAs, prediction metadata
+
+</details>
+
 ### MultiQC report
 
 <details markdown="1">

docs/usage.md

@@ -39,7 +39,7 @@ Each FASTA file should contain a single protein sequence unless using multimer m
 
 ## Running the pipeline
 
-The typical commands for running the pipeline on AlphaFold2, Colabfold and ESMFold modes are shown below.
+The typical commands for running the pipeline on AlphaFold2, Colabfold, ESMFold and RoseTTAFold-All-Atom modes are shown below.
 
 > You can run any combination of the models by providing them to the `--mode` parameter separated by a comma. For example: `--mode alphafold2,esmfold,colabfold` will run the three models in parallel.
 
@@ -428,6 +428,18 @@ If you specify the `--esmfold_db <PATH>` parameter, the directory structure of y
 
 This will launch the pipeline with the `docker` configuration profile. See below for more information about profiles.
 
+AlphaFold2 regular can be run using this command:
+
+```bash
+nextflow run nf-core/proteinfold \
+      --input samplesheet.csv \
+      --outdir <OUTDIR> \
+      --mode rosettafold_all_atom \
+      --rosettafold_all_atom_db <null (default) | DB_PATH> \
+      --use_gpu <true/false> \
+      -profile <docker/singularity/.../institute>
+```
+
 Note that the pipeline will create the following files in your working directory:
 
 ```bash