From c447c45d695e61a07cb3ed7eea4920f87e8cda74 Mon Sep 17 00:00:00 2001
From: Mira Kuntz <mira.kuntz@yahoo.com>
Date: Tue, 7 May 2024 13:52:01 +0200
Subject: [PATCH] give 10% of alphafold jobs gpu access

---
 files/galaxy/tpv/tools.yml | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/files/galaxy/tpv/tools.yml b/files/galaxy/tpv/tools.yml
index 7925c16cd..099888be7 100644
--- a/files/galaxy/tpv/tools.yml
+++ b/files/galaxy/tpv/tools.yml
@@ -284,6 +284,10 @@ tools:
       - if: helpers.tool_version_gte(tool, '2.3.1+galaxy2')
         params:
           singularity_run_extra_arguments: "--env ALPHAFOLD_DB=/data/db/databases/alphafold_databases,ALPHAFOLD_USE_GPU=False"
+        execute: |
+          from random import SystemRandom
+          if SystemRandom().random() < 0.10:
+              entity.gpus = 1
         # tweak amount of requested memory depending on the AlphaFold model to be run
       - id: model_preset_multimer
         if: job.get_param_values(app).get("model_preset") == "multimer"