#!/bin/bash
# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
set -e
if [ $(basename ${PWD}) != "src" ]; then
echo "Please set the current working directory to chromium/src first!"
exit 1
fi
files=(
"LICENSE"
"tensorflow_text/core/kernels/regex_split.cc"
"tensorflow_text/core/kernels/regex_split.h"
"tensorflow_text/core/kernels/wordpiece_tokenizer.cc"
"tensorflow_text/core/kernels/wordpiece_tokenizer.h"
)
git clone --depth 1 https://github.com/tensorflow/text /tmp/text
rm -rf third_party/tensorflow-text/src/*
pushd third_party/tensorflow-text/src/
for file in ${files[@]} ; do
if [ ! -d "$(dirname ${file})" ] ; then
mkdir -p "$(dirname ${file})"
fi
cp "/tmp/text/${file}" "${file}"
done
popd
rm -rf /tmp/text