Fix Ruff.

aryn-ai · Oct 6, 2024 · d4bd335 · d4bd335
1 parent dd00606
commit d4bd335
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git 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
@@ -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