summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/py-tensorflow-datasets/package.py37
-rw-r--r--var/spack/repos/builtin/packages/py-tensorflow-metadata/package.py36
2 files changed, 73 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-tensorflow-datasets/package.py b/var/spack/repos/builtin/packages/py-tensorflow-datasets/package.py
new file mode 100644
index 0000000000..5a76e3c433
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-tensorflow-datasets/package.py
@@ -0,0 +1,37 @@
+# Copyright 2013-2021 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 import *
+
+
+class PyTensorflowDatasets(PythonPackage):
+ """tensorflow/datasets is a library of datasets ready to use with
+ TensorFlow."""
+
+ homepage = "https://github.com/tensorflow/datasets"
+ pypi = "tensorflow-datasets/tensorflow-datasets-4.4.0.tar.gz"
+
+ version(
+ "4.4.0",
+ sha256="3e95a61dec1fdb7b05dabc0dbed1b531e13d6c6fd362411423d0a775e5e9b960",
+ )
+
+ depends_on("python@3.6:", type=("build", "run"))
+ depends_on("py-setuptools", type="build")
+ depends_on("py-absl-py", type=("build", "run"))
+ depends_on("py-attrs@18.1.0:", type=("build", "run"))
+ depends_on("py-dill", type=("build", "run"))
+ depends_on("py-future", type=("build", "run"))
+ depends_on("py-numpy", type=("build", "run"))
+ depends_on("py-promise", type=("build", "run"))
+ depends_on("py-protobuf@3.12.2:", type=("build", "run"))
+ depends_on("py-requests@2.19.0:", type=("build", "run"))
+ depends_on("py-six", type=("build", "run"))
+ depends_on("py-tensorflow-metadata", type=("build", "run"))
+ depends_on("py-termcolor", type=("build", "run"))
+ depends_on("py-tqdm", type=("build", "run"))
+ depends_on("py-dataclasses", type=("build", "run"), when="python@:3.6")
+ depends_on("py-typing-extensions", type=("build", "run"), when="python@:3.7")
+ depends_on("py-importlib-resources", type=("build", "run"), when="python@:3.8")
diff --git a/var/spack/repos/builtin/packages/py-tensorflow-metadata/package.py b/var/spack/repos/builtin/packages/py-tensorflow-metadata/package.py
new file mode 100644
index 0000000000..e7d489dbdb
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-tensorflow-metadata/package.py
@@ -0,0 +1,36 @@
+# Copyright 2013-2021 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)
+
+import tempfile
+
+from spack import *
+
+
+class PyTensorflowMetadata(PythonPackage):
+ """Library and standards for schema and statistics.
+
+ TensorFlow Metadata provides standard representations for metadata that are
+ useful when training machine learning models with TensorFlow."""
+
+ homepage = "https://pypi.org/project/tensorflow-metadata/"
+
+ # Only available as a wheel on PyPI
+ url = "https://github.com/tensorflow/metadata/archive/refs/tags/v1.5.0.tar.gz"
+
+ version(
+ "1.5.0",
+ sha256="f0ec8aaf62fd772ef908efe4ee5ea3bc0d67dcbf10ae118415b7b206a1d61745",
+ )
+
+ depends_on("bazel@0.24.1:", type="build")
+ depends_on("python@3.7:3", type=("build", "run"))
+ depends_on("py-setuptools", type="build")
+ depends_on("py-absl-py@0.9:0.12", type=("build", "run"))
+ depends_on("py-googleapis-common-protos@1.52.0:1", type=("build", "run"))
+ depends_on("py-protobuf@3.13:3", type=("build", "run"))
+
+ def setup_build_environment(self, env):
+ tmp_path = tempfile.mkdtemp(prefix="spack")
+ env.set("TEST_TMPDIR", tmp_path)