Now that YALTAi can have multiple formats, account for this

PonteIneptique · Oct 18, 2024 · 62270d4 · 62270d4
1 parent 64ffaca
commit 62270d4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/rtk/task.py b/rtk/task.py
@@ -543,7 +543,7 @@ def __init__(
         if not os.path.exists(yolo_model):
             raise ValueError(f"Unknown YOLOv8 model `{yolo_model}`")
 
-        cmd = (f"{binary} kraken --verbose "
+        cmd = (f"{binary} kraken --verbose --alto "
                f"{' --raise-on-error ' if kwargs.get('raise_on_error') else ''} --device {device} R "
                f"segment -y {yolo_model}").split(" ")
 

diff --git a/tests/test_yaltai.py b/tests/test_yaltai.py
@@ -15,3 +15,6 @@ def test_yaltai(capsys):
     yaltai.process()
     assert len(yaltai.output_files) == 2, "Two files were processed"
     assert sorted(yaltai.output_files) == ['test_output/page1.xml', 'test_output/page2.xml'], "Two files were processed"
+    with open('test_output/page1.xml') as f:
+        data = f.read()
+    assert "<alto" in data