lint

Author: tonylampada

tests/util/test_folderparser.py

@@ -58,10 +58,12 @@ def test_paligemma_format(self):
         testImagePath = "/dataset/de48275e1ff70fab78bee31e09fc896d_png.rf.01a97b1ad053aa1e6525ac0451cee8b7.jpg"
         testImage = [i for i in parsed["images"] if i["file"] == testImagePath][0]
         assert testImage["annotationfile"]["name"] == "annotation.jsonl"
-        # fmt: off
-        expected = """{"image": "de48275e1ff70fab78bee31e09fc896d_png.rf.01a97b1ad053aa1e6525ac0451cee8b7.jpg", "prefix": "Which sector had the highest ROI in 2013?", "suffix": "Retail"}
-{"image": "de48275e1ff70fab78bee31e09fc896d_png.rf.01a97b1ad053aa1e6525ac0451cee8b7.jpg", "prefix": "Which sector had the highest ROI in 2014?", "suffix": "Electronics"}"""
-        # fmt: on
+        expected = (
+            '{"image": "de48275e1ff70fab78bee31e09fc896d_png.rf.01a97b1ad053aa1e6525ac0451cee8b7.jpg",'
+            ' "prefix": "Which sector had the highest ROI in 2013?", "suffix": "Retail"}\n'
+            '{"image": "de48275e1ff70fab78bee31e09fc896d_png.rf.01a97b1ad053aa1e6525ac0451cee8b7.jpg",'
+            ' "prefix": "Which sector had the highest ROI in 2014?", "suffix": "Electronics"}'
+        )
         assert testImage["annotationfile"]["rawText"] == expected