Skip to content

Commit 56b04d4

Browse files
committed
Added the same --tmp_dir flag to tesstrain_utils.sh
1 parent 25fa392 commit 56b04d4

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/training/tesstrain_utils.sh

+8-1
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,9 @@ parse_flags() {
149149
--fonts_dir)
150150
parse_value "FONTS_DIR" ${ARGV[$j]:-}
151151
i=$j ;;
152+
--tmp_dir)
153+
parse_value "TMP_DIR" ${ARGV[$j]:-}
154+
i=$j ;;
152155
--lang)
153156
parse_value "LANG_CODE" ${ARGV[$j]:-}
154157
i=$j ;;
@@ -215,7 +218,11 @@ parse_flags() {
215218

216219
# Location where intermediate files will be created.
217220
TIMESTAMP=$(date +%Y-%m-%d)
218-
TMP_DIR=$(mktemp -d -t ${LANG_CODE}-${TIMESTAMP}.XXX)
221+
if [[ -z ${TMP_DIR:-} ]]; then
222+
TMP_DIR=$(mktemp -d -t ${LANG_CODE}-${TIMESTAMP}.XXX)
223+
else
224+
TMP_DIR=$(mktemp -d -p ${TMP_DIR} -t ${LANG_CODE}-${TIMESTAMP}.XXX)
225+
fi
219226
TRAINING_DIR=${TMP_DIR}
220227
# Location of log file for the whole run.
221228
LOG_FILE=${TRAINING_DIR}/tesstrain.log

0 commit comments

Comments
 (0)