114 lines
4.1 KiB
Plaintext
114 lines
4.1 KiB
Plaintext
#!/usr/bin/env nextflow
|
|
|
|
nextflow.enable.dsl=2
|
|
|
|
// Pipeline parameters
|
|
params.pdb = 's3://omic/eureka/Pocketminer/1HSG.pdb'
|
|
params.outdir = 's3://omic/eureka/Pocketminer/output'
|
|
params.model_path = '/workspace/gvp/models/pocketminer'
|
|
params.debug = false
|
|
|
|
process POCKETMINER {
|
|
container 'harbor.cluster.omic.ai/omic/pocketminer:latest'
|
|
publishDir params.outdir, mode: 'copy'
|
|
|
|
input:
|
|
val pdb_path
|
|
|
|
output:
|
|
path "*-preds.npy", emit: predictions_npy
|
|
path "*-predictions.txt", emit: predictions_txt
|
|
path "*-summary.json", emit: summary
|
|
path "*_X.npy", optional: true, emit: features_debug
|
|
path "*_S.npy", optional: true, emit: sequence_debug
|
|
path "*_mask.npy", optional: true, emit: mask_debug
|
|
path "run.log", emit: log
|
|
|
|
script:
|
|
def debug_flag = params.debug ? '--debug' : ''
|
|
"""
|
|
set +e
|
|
touch run.log
|
|
|
|
echo "=== Environment Debug ===" >> run.log
|
|
echo "Input path: ${pdb_path}" >> run.log
|
|
echo "AWS_ENDPOINT_URL: \${AWS_ENDPOINT_URL:-not set}" >> run.log
|
|
echo "AWS_ACCESS_KEY_ID set: \${AWS_ACCESS_KEY_ID:+yes}" >> run.log
|
|
ls /omic/eureka/ >> run.log 2>&1 || echo "/omic/eureka not mounted" >> run.log
|
|
echo "=== End Debug ===" >> run.log
|
|
|
|
PDB_FILE=""
|
|
|
|
# Method 1: Direct path (if PVC mounted or local)
|
|
if [ -f "${pdb_path}" ]; then
|
|
echo "Found at direct path" >> run.log
|
|
cp "${pdb_path}" input.pdb && PDB_FILE="input.pdb"
|
|
fi
|
|
|
|
# Method 2: Convert s3://omic/... to /omic/... (PVC mount path)
|
|
if [ -z "\$PDB_FILE" ]; then
|
|
PVC_PATH="\$(echo '${pdb_path}' | sed 's|^s3://|/|')"
|
|
if [ -f "\$PVC_PATH" ]; then
|
|
echo "Found at PVC path: \$PVC_PATH" >> run.log
|
|
cp "\$PVC_PATH" input.pdb && PDB_FILE="input.pdb"
|
|
else
|
|
echo "Not found at PVC path: \$PVC_PATH" >> run.log
|
|
fi
|
|
fi
|
|
|
|
# Method 3: Download from MinIO via curl (no auth needed for public)
|
|
if [ -z "\$PDB_FILE" ]; then
|
|
S3_PATH="\$(echo '${pdb_path}' | sed 's|^s3://||')"
|
|
MINIO_URL="http://datalake-hl.datalake.svc.cluster.local:9000/\$S3_PATH"
|
|
echo "Downloading from MinIO: \$MINIO_URL" >> run.log
|
|
curl -sf "\$MINIO_URL" -o input.pdb 2>> run.log
|
|
if [ -f input.pdb ] && [ -s input.pdb ]; then
|
|
echo "Downloaded from MinIO (\$(wc -c < input.pdb) bytes)" >> run.log
|
|
PDB_FILE="input.pdb"
|
|
else
|
|
echo "MinIO download failed or empty" >> run.log
|
|
rm -f input.pdb
|
|
fi
|
|
fi
|
|
|
|
# Method 4: Download from MinIO using AWS env vars
|
|
if [ -z "\$PDB_FILE" ] && [ -n "\${AWS_ACCESS_KEY_ID:-}" ]; then
|
|
S3_PATH="\$(echo '${pdb_path}' | sed 's|^s3://||')"
|
|
ENDPOINT="\${AWS_ENDPOINT_URL:-http://datalake-hl.datalake.svc.cluster.local:9000}"
|
|
echo "Downloading with AWS creds from: \$ENDPOINT/\$S3_PATH" >> run.log
|
|
curl -sf -u "\${AWS_ACCESS_KEY_ID}:\${AWS_SECRET_ACCESS_KEY}" "\$ENDPOINT/\$S3_PATH" -o input.pdb 2>> run.log
|
|
if [ -f input.pdb ] && [ -s input.pdb ]; then
|
|
echo "Downloaded with AWS creds (\$(wc -c < input.pdb) bytes)" >> run.log
|
|
PDB_FILE="input.pdb"
|
|
else
|
|
echo "AWS cred download failed" >> run.log
|
|
rm -f input.pdb
|
|
fi
|
|
fi
|
|
|
|
set -e
|
|
|
|
if [ -z "\$PDB_FILE" ] || [ ! -f input.pdb ]; then
|
|
echo "ERROR: Could not retrieve PDB file from any source" >> run.log
|
|
cat run.log
|
|
exit 1
|
|
fi
|
|
|
|
echo "PDB file ready (\$(wc -c < input.pdb) bytes)" >> run.log
|
|
PDB_BASENAME="\$(basename '${pdb_path}' .pdb)"
|
|
|
|
python /workspace/entrypoint.py \\
|
|
--pdb input.pdb \\
|
|
--output-folder . \\
|
|
--output-name "\$PDB_BASENAME" \\
|
|
--model-path ${params.model_path} \\
|
|
${debug_flag} 2>&1 | tee -a run.log
|
|
|
|
echo "Pipeline completed successfully" >> run.log
|
|
"""
|
|
}
|
|
|
|
workflow {
|
|
POCKETMINER(params.pdb)
|
|
}
|