summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/py-exarl/package.py39
1 files changed, 39 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-exarl/package.py b/var/spack/repos/builtin/packages/py-exarl/package.py
new file mode 100644
index 0000000000..c3106e8778
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-exarl/package.py
@@ -0,0 +1,39 @@
+# Copyright 2013-2023 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 PyExarl(PythonPackage):
+ """A scalable software framework for reinforcement learning environments
+ and agents/policies used for the Design and Control applications"""
+
+ homepage = "https://github.com/exalearn/EXARL"
+ git = "https://github.com/exalearn/EXARL.git"
+
+ maintainers("cmahrens")
+
+ version("master", branch="master")
+ version("develop", branch="develop")
+ version("update-spack", branch="update-spack")
+ version("0.1.0", tag="v0.1.0")
+
+ depends_on("python@3.6:", type=("build", "run"))
+ depends_on("git-lfs", type=("build"))
+ depends_on("py-setuptools", type=("build"))
+
+ depends_on("py-ase", type=("build", "run"))
+ depends_on("py-numba", type=("build", "run"))
+ depends_on("py-lmfit", type=("build", "run"))
+ depends_on("py-seaborn", type=("build", "run"))
+ depends_on("py-keras", type=("build", "run"))
+ depends_on("py-mpi4py", type=("build", "run"))
+ depends_on("py-scikit-learn", type=("build", "run"))
+ depends_on("py-tensorflow", type=("build", "run"))
+ depends_on("py-torch", type=("build", "run"))
+ depends_on("py-torchvision", type=("build", "run"))
+
+ phases = ["install"]