From 119e7808603187b327fd7bc58ec3662bbf8e9359 Mon Sep 17 00:00:00 2001
From: marmaduke <marmaduke.woodman@univ-amu.fr>
Date: Thu, 20 Mar 2025 11:23:55 +0000
Subject: [PATCH] feat(py-vbjax): add package

---
 packages/py-vbjax/package.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 packages/py-vbjax/package.py

diff --git a/packages/py-vbjax/package.py b/packages/py-vbjax/package.py
new file mode 100644
index 00000000..53c7e034
--- /dev/null
+++ b/packages/py-vbjax/package.py
@@ -0,0 +1,26 @@
+# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+from spack.package import *
+
+
+class PyVbjax(PythonPackage):
+    """Virtual brain models in JAX."""
+
+    homepage = "https://github.com/ins-amu/vbjax"
+    pypi = "vbjax/vbjax-0.0.16.tar.gz"
+    git = "https://github.com/ins-amu/vbjax"
+
+    # 'joblib.test' requires 'pytest'. Leave out of 'import_modules' to avoid
+    # unnecessary dependencies.
+    skip_modules = ["joblib.test"]
+
+    license("Apache-2.0")
+
+    version("0.0.16")
+
+    depends_on('py-jax', type=('build', 'link', 'run'))
+    depends_on("python", type=("build", "run"))
+    depends_on("py-setuptools", type=("build", "run"))
-- 
GitLab