mirror of
https://github.com/Unstructured-IO/unstructured.git
synced 2025-07-12 11:35:53 +00:00
01a0e003d9
### Summary A follow up ticket on https://github.com/Unstructured-IO/unstructured/pull/1801, I forgot to remove the lines that pass extract_tables to inference, and noted the table regression if we only do one OCR for entire doc **Tech details:** * stop passing `extract_tables` parameter to inference * added table extraction ingest test for image, which was skipped before, and the "text_as_html" field contains the OCR output from the table OCR refactor PR * replaced `assert_called_once_with` with `call_args` so that the unit tests don't need to test additional parameters * added `error_margin` as ENV when comparing bounding boxes of`ocr_region` with `table_element` * added more tests for tables and noted the table regression in test for partition pdf ### Test * for stop passing `extract_tables` parameter to inference, run test `test_partition_pdf_hi_res_ocr_mode_with_table_extraction` before this branch and you will see warning like `Table OCR from get_tokens method will be deprecated....`, which means it called the table OCR in inference repo. This branch removed the warning.
59 lines
2.9 KiB
Bash
Executable File
59 lines
2.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# Structured .json output from PDF's or images may differ subtly (or not so subtly)
|
|
# based on the version of tesseract, its dependencies, and chip architecture.
|
|
#
|
|
# To update ingest-test expected outputs (structured .json files), this script:
|
|
# * builds an ubuntu image that
|
|
# * matches CI with respect to tesseract and OS deps
|
|
# * installs python dependencies from the local requirements/ directory
|
|
# * runs each test ingest script with OVERWRITE_FIXTURES=true
|
|
# * so updates are written to test_unstructured_ingest/expected-structured-output/
|
|
# * using local unstructured/ directory (i.e. from local git branch)
|
|
#
|
|
# It is recommended to run this script on x86_64 hardware.
|
|
|
|
set -eu -o pipefail
|
|
|
|
# Change to the root of the repository
|
|
SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
|
|
cd "$SCRIPT_DIR"/.. || exit 1
|
|
|
|
ARCHITECTURE=$(uname -m)
|
|
|
|
if [ "$ARCHITECTURE" != "x86_64" ]; then
|
|
echo "Warning: This script is designed to run on x86_64 hardware, but you're running on $ARCHITECTURE."
|
|
fi
|
|
|
|
./scripts/docker-build-ubuntu.sh
|
|
|
|
# Warn the user if they have an old image
|
|
IMAGE_NAME="unstructured-ubuntu:latest"
|
|
CREATION_TIMESTAMP=$(docker inspect --format='{{.Created}}' "$IMAGE_NAME")
|
|
CREATION_DATE=$(date -d "$CREATION_TIMESTAMP" +%s)
|
|
CURRENT_DATE=$(date +%s)
|
|
AGE_DAYS=$(( (CURRENT_DATE - CREATION_DATE) / 86400 ))
|
|
if [ "$AGE_DAYS" -gt 6 ]; then
|
|
echo "WARNING: The image \"$IMAGE_NAME\" is more than 7 days old ($AGE_DAYS days)."
|
|
echo "You may want to 'docker rmi $IMAGE_NAME' and rerun this script if it is not current."
|
|
fi
|
|
|
|
docker run --rm -v "$SCRIPT_DIR"/../unstructured:/root/unstructured \
|
|
-v "$SCRIPT_DIR"/../test_unstructured_ingest:/root/test_unstructured_ingest \
|
|
${DISCORD_TOKEN:+-e DISCORD_TOKEN="$DISCORD_TOKEN"} \
|
|
${SLACK_TOKEN:+-e SLACK_TOKEN="$SLACK_TOKEN"} \
|
|
${CONFLUENCE_USER_EMAIL:+-e CONFLUENCE_USER_EMAIL="$CONFLUENCE_USER_EMAIL"} \
|
|
${CONFLUENCE_API_TOKEN:+-e CONFLUENCE_API_TOKEN="$CONFLUENCE_API_TOKEN"} \
|
|
${GH_READ_ONLY_ACCESS_TOKEN:+-e GH_READ_ONLY_ACCESS_TOKEN="$GH_READ_ONLY_ACCESS_TOKEN"} \
|
|
-w /root "$IMAGE_NAME" \
|
|
bash -c "export OVERWRITE_FIXTURES=true && source ~/.bashrc && pyenv activate unstructured && tesseract --version &&
|
|
./test_unstructured_ingest/test-ingest-azure.sh &&
|
|
./test_unstructured_ingest/test-ingest-discord.sh &&
|
|
./test_unstructured_ingest/test-ingest-github.sh &&
|
|
./test_unstructured_ingest/test-ingest-biomed-api.sh &&
|
|
./test_unstructured_ingest/test-ingest-biomed-path.sh &&
|
|
./test_unstructured_ingest/test-ingest-s3.sh &&
|
|
./test_unstructured_ingest/test-ingest-slack.sh &&
|
|
./test_unstructured_ingest/test-ingest-pdf-fast-reprocess.sh &&
|
|
./test_unstructured_ingest/test-ingest-local-single-file-with-pdf-infer-table-structure.sh"
|