Lint and format

scripts/alpaca_json_to_jsonl.py

@@ -1,23 +1,39 @@
+"""Module to convert json file to jsonl"""
+
 import os
 import sys
+
+from typing import Optional
 from pathlib import Path
 
 import fire
-from typing import Optional
+
+
+from axolotl.convert import (
+    FileReader,
+    StdoutWriter,
+    FileWriter,
+    JsonlSerializer,
+    JsonParser,
+    JsonToJsonlConverter,
+)
+
 
 # add src to the pythonpath so we don't need to pip install this
 project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
 src_dir = os.path.join(project_root, "src")
 sys.path.insert(0, src_dir)
 
-from axolotl.convert import *
-
 
 def main(
-    input: Path,
+    file: Path,
     output: Optional[Path] = None,
     to_stdout: Optional[bool] = False,
 ):
+    """
+    Convert a json file to jsonl
+    """
+
     file_reader = FileReader()
     if to_stdout or output is None:
         writer = StdoutWriter()
@@ -28,7 +44,7 @@ def main(
 
     converter = JsonToJsonlConverter(file_reader, writer, json_parser, jsonl_serializer)
 
-    converter.convert(input, output)
+    converter.convert(file, output)
 
 
 if __name__ == "__main__":