25 lines
458 B
Bash
Executable File
25 lines
458 B
Bash
Executable File
set -e
|
|
set -x
|
|
SRCDIR=`dirname $0`
|
|
|
|
|
|
on_error () {
|
|
rm -fr $workdir
|
|
rm -rf $SRCDIR/torch-shm-file-*
|
|
}
|
|
|
|
# allow faster local testing
|
|
if test -d $(dirname ${SRCDIR})/.embeddings; then
|
|
EMBEDDING_DIR="$(dirname ${SRCDIR})/.embeddings"
|
|
else
|
|
mkdir -p $SRCDIR/embeddings
|
|
EMBEDDING_DIR="$SRCDIR/embeddings"
|
|
fi
|
|
|
|
export SENTENCE_TRANSFORMERS_HOME="$EMBEDDING_DIR"
|
|
|
|
TMPDIR=`pwd`
|
|
workdir=`mktemp -d $TMPDIR/genieNLP-tests-XXXXXX`
|
|
trap on_error ERR INT TERM
|
|
|