mirror of
https://github.com/Unstructured-IO/unstructured.git
synced 2025-08-20 06:39:41 +00:00
81af879038
### Description In an effort to speed up the ingest tests, bumping the num if processes to the max on the system for each
45 lines
1.6 KiB
Bash
Executable File
45 lines
1.6 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
set -e
|
|
|
|
SCRIPT_DIR=$(dirname "$(realpath "$0")")
|
|
cd "$SCRIPT_DIR"/.. || exit 1
|
|
OUTPUT_FOLDER_NAME=github
|
|
OUTPUT_DIR=$SCRIPT_DIR/structured-output/$OUTPUT_FOLDER_NAME
|
|
DOWNLOAD_DIR=$SCRIPT_DIR/download/$OUTPUT_FOLDER_NAME
|
|
max_processes=${MAX_PROCESSES:=$(python -c "import os; print(os.cpu_count())")}
|
|
|
|
# shellcheck disable=SC1091
|
|
source "$SCRIPT_DIR"/cleanup.sh
|
|
trap 'cleanup_dir "$OUTPUT_DIR"' EXIT
|
|
|
|
GH_READ_ONLY_ACCESS_TOKEN=${GH_READ_ONLY_ACCESS_TOKEN:-none}
|
|
|
|
ACCESS_TOKEN_FLAGS=""
|
|
# to update test fixtures, "export OVERWRITE_FIXTURES=true" and rerun this script
|
|
if [[ "$GH_READ_ONLY_ACCESS_TOKEN" != "none" ]]; then
|
|
ACCESS_TOKEN_FLAGS="--git-access-token $GH_READ_ONLY_ACCESS_TOKEN"
|
|
elif [[ "$CI" == "true" ]]; then
|
|
echo "Warning: GH_READ_ONLY_ACCESS_TOKEN is not defined in the CI environment."
|
|
echo "This can lead to intermittent failures in test-ingest-github.sh, as non-auth'ed"
|
|
echo "requests are severely rate limited by GitHub."
|
|
echo
|
|
fi
|
|
|
|
#shellcheck disable=SC2086
|
|
PYTHONPATH=. ./unstructured/ingest/main.py \
|
|
github \
|
|
--num-processes "$max_processes" \
|
|
--download-dir "$DOWNLOAD_DIR" \
|
|
--metadata-exclude coordinates,filename,file_directory,metadata.data_source.date_processed,metadata.last_modified,metadata.detection_class_prob,metadata.parent_id,metadata.category_depth \
|
|
--strategy hi_res \
|
|
--preserve-downloads \
|
|
--reprocess \
|
|
--output-dir "$OUTPUT_DIR" \
|
|
--verbose \
|
|
--url dcneiner/Downloadify \
|
|
--git-file-glob '*.html,*.txt' \
|
|
$ACCESS_TOKEN_FLAGS
|
|
|
|
"$SCRIPT_DIR"/check-diff-expected-output.sh $OUTPUT_FOLDER_NAME
|