testing openproblems image

Author: ghar1821

src/metrics/bras/config.vsh.yaml

@@ -68,24 +68,12 @@ engines:
   # Specifications for the Docker image for this component.
   # testing gpu jax version
   - type: docker
-    image: nvidia/cuda:12.4.0-runtime-ubuntu22.04
+    image: openproblems/base_pytorch_nvidia:1.1
     setup:
-      - type: apt
-        packages:
-          - python3-pip
-          - procps
-          - git
       - type: python
         packages:
           - jax[cuda_12_pip]
-          - anndata~=0.11.0
-          - scanpy~=1.11.0
           - scib-metrics~=0.5.7
-          - pyyaml
-          - requests
-          - jsonschema
-        github:
-          - openproblems-bio/core#subdirectory=packages/python/openproblems
   # - type: docker
   #   image: python:3.11
   #   setup: