Update build script and test

neuml · Dec 3, 2024 · 92489ac · 92489ac
1 parent b05abd2
commit 92489ac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -34,7 +34,6 @@ jobs:
         run: |
           echo "OMP_NUM_THREADS=1" >> $GITHUB_ENV
           echo "PYTORCH_MPS_DISABLE=1" >> $GITHUB_ENV
-          echo "PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0" >> $GITHUB_ENV
           echo "LLAMA_NO_METAL=1" >> $GITHUB_ENV
           echo "TIKA_STARTUP_SLEEP=30" >> $GITHUB_ENV
           echo "TIKA_STARTUP_MAX_RETRY=10" >> $GITHUB_ENV

diff --git a/test/python/testpipeline/testdata/testtextractor.py b/test/python/testpipeline/testdata/testtextractor.py
@@ -2,10 +2,9 @@
 Textractor module tests
 """
 
+import platform
 import unittest
 
-import torch
-
 from txtai.pipeline import Textractor
 
 # pylint: disable = C0411
@@ -51,14 +50,12 @@ def testDefault(self):
         # Check number of sections is as expected
         self.assertEqual(len(sections), 2)
 
+    @unittest.skipIf(platform.system() == "Darwin", "Docling skipped on macOS to avoid MPS issues")
     def testDocling(self):
         """
         Test docling backend
         """
 
-        # Set default device to CPU
-        torch.set_default_device("cpu")
-
         textractor = Textractor(backend="docling")
 
         # Extract text and check for Markdown formatting