From d4bd335689dd6656b517263658d53a743b4eac71 Mon Sep 17 00:00:00 2001 From: Matt Welsh Date: Sun, 6 Oct 2024 16:44:49 -0700 Subject: [PATCH] Fix Ruff. --- apps/query-demo/querydemo/loaddata.py | 5 ++--- apps/query-ui/queryui/loaddata.py | 1 - 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/apps/query-demo/querydemo/loaddata.py b/apps/query-demo/querydemo/loaddata.py index 565a28d09..45face4aa 100644 --- a/apps/query-demo/querydemo/loaddata.py +++ b/apps/query-demo/querydemo/loaddata.py @@ -20,7 +20,6 @@ OpenAIPropertyExtractor, ) from sycamore.transforms.merge_elements import GreedyTextElementMerger -from sycamore.transforms.summarize_images import SummarizeImages from sycamore.llms import OpenAI, OpenAIModels from sycamore.transforms.embed import SentenceTransformerEmbedder from opensearchpy import OpenSearch @@ -148,8 +147,8 @@ def main(): partitioned_docset = ( docset.partition(partitioner=ArynPartitioner(extract_table_structure=True, use_ocr=True, extract_images=True)) -# XXX MDW 3 Oct 2024 - Disable this for now as it seems to be failing on the demo instance. -# .transform(SummarizeImages) + # XXX MDW 3 Oct 2024 - Disable this for now as it seems to be failing on the demo instance. + # .transform(SummarizeImages) .materialize(path=f"{args.tempdir}/ntsb-loader-stage-0", source_mode=sycamore.MATERIALIZE_USE_STORED) .map(add_schema_property) .materialize(path=f"{args.tempdir}/ntsb-loader-stage-1", source_mode=sycamore.MATERIALIZE_USE_STORED) diff --git a/apps/query-ui/queryui/loaddata.py b/apps/query-ui/queryui/loaddata.py index e2f11a2d9..37c6835e8 100644 --- a/apps/query-ui/queryui/loaddata.py +++ b/apps/query-ui/queryui/loaddata.py @@ -21,7 +21,6 @@ OpenAIPropertyExtractor, ) from sycamore.transforms.merge_elements import GreedyTextElementMerger -from sycamore.transforms.summarize_images import SummarizeImages from sycamore.llms import OpenAI, OpenAIModels from sycamore.transforms.embed import SentenceTransformerEmbedder from opensearchpy import OpenSearch