new package: py-tensorflow-datasets (#27746)

* new package: py-tensorflow-datasets

- includes new dependency package: py-tensorflow-metadata

* Update var/spack/repos/builtin/packages/py-tensorflow-datasets/package.py

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>

* Update var/spack/repos/builtin/packages/py-tensorflow-metadata/package.py

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>

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)