Add tensorflow-text package

recipes/tensorflow-text/build.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+set -exuo pipefail
+
+export HERMETIC_PYTHON_VERSION=${python}
+
+WHEEL_DIR=${PWD}/wheel_dir
+mkdir -p ${WHEEL_DIR}
+bazel build ./oss_scripts/pip_package:build_pip_package
+bazel-bin/text/oss_scripts/pip_package/build_pip_package ${WHEEL_DIR}
+${PYTHON} -m pip install --no-deps ${WHEEL_DIR}/*.whl
+bazel clean

recipes/tensorflow-text/meta.yaml

@@ -0,0 +1,61 @@
+{% set version = "2.17.0" %}
+
+package:
+  name: tensorflow-text
+  version: {{ version }}
+
+source:
+  url: https://github.com/tensorflow/text/archive/v{{ version }}.tar.gz
+  sha256: 039f05f422a3ad55289a4897b5427d07cac7efe3943927c3b3c7a593f6123989
+
+build:
+  number: 0
+  skip: true  # [win]
+  string: cuda{{ cuda_compiler_version | replace('.', '') }}py{{ CONDA_PY }}h{{ PKG_HASH }}_{{ PKG_BUILDNUM }}  # [cuda_compiler_version != "None"]
+  string: cpu_py{{ CONDA_PY }}h{{ PKG_HASH }}_{{ PKG_BUILDNUM }}  # [cuda_compiler_version == "None"]
+
+requirements:
+  build:
+  - {{ compiler('c') }}
+  - {{ stdlib('c') }}
+  - {{ compiler('cxx') }}
+  - {{ compiler('cuda') }}                 # [cuda_compiler_version != "None"]
+  - bazel >=5.2.0
+  host:
+  - python
+  - pip
+  - packaging
+  - setuptools >=70
+  - wheel
+  - absl-py
+  - auditwheel
+  - patchelf
+  # This ensures that a consistent version of openssl is chosen between
+  # all packages.
+  # https://github.com/conda-forge/conda-forge.github.io/issues/1528
+  - openssl
+  - {{ pin_compatible('tensorflow', max_pin='x.x') }}
+  run:
+  - python
+  - {{ pin_compatible('tensorflow', max_pin='x.x') }}
+test:
+  requires:
+  - pip
+  - absl-py
+  - pytest
+  - tf-keras
+  - tensorflow-datasets >=3.2.0
+  commands:
+  - pip check
+
+about:
+  home: https://www.tensorflow.org/text/
+  license: Apache-2.0
+  license_family: Apache
+  license_file: LICENSE
+  summary: TF.Text is a TensorFlow library of text related ops, modules, and subgraphs.
+  dev_url: https://github.com/tensorflow/text
+
+extra:
+  recipe-maintainers:
+  - ehfd