add flax dependency

LeonHafner · LeonHafner · commit 997a9114b22f · 2025-02-27T14:43:04.000+01:00
diff --git a/src/methods_cell_type_annotation/moscot/config.vsh.yaml b/src/methods_cell_type_annotation/moscot/config.vsh.yaml
@@ -50,7 +50,7 @@ engines:
     image: openproblems/base_python:1.0.0
     setup:
       - type: python
-        pypi: [numpy, anndata, scanpy, moscot]
+        pypi: [numpy, anndata, scanpy, moscot, flax]
   - type: native
 
 runners:
