cleanup

Signed-off-by: Mandana Vaziri <[email protected]>

Author: vazirim

src/pdl/pdl_interpreter.py

@@ -1,5 +1,5 @@
 # pylint: disable=import-outside-toplevel
-from io import StringIO
+import csv
 import json
 import re
 import shlex
@@ -8,13 +8,13 @@
 import time
 import traceback
 import types
-import csv
 
 # TODO: temporarily disabling warnings to mute a pydantic warning from liteLLM
 import warnings
 from abc import ABC, abstractmethod
 from concurrent.futures import ThreadPoolExecutor
 from functools import partial, reduce
+from io import StringIO
 from itertools import count
 from os import getenv
 from pathlib import Path

tests/test_parser.py

@@ -173,6 +173,7 @@ def test_parser_case2():
     result = exec_str(prog)
     assert result == ["1", "2", "3", "4"]
 
+
 def test_parser_csv():
     csv_parser = """
     text: |
@@ -182,4 +183,8 @@ def test_parser_csv():
     parser: csv
     """
     result = exec_str(csv_parser)
-    assert result == [["1", "Apple", "Red"], ["2", "Orange", "Orange"], ["3", "Banana", "Yellow"]]
+    assert result == [
+        ["1", "Apple", "Red"],
+        ["2", "Orange", "Orange"],
+        ["3", "Banana", "Yellow"],
+    ]