cli fix

jessekrubin · Nov 2, 2023 · 512120b · 512120b
1 parent d67a994
commit 512120b
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/python/utiles/cli.py b/python/utiles/cli.py
@@ -297,9 +297,19 @@ def feature_gen() -> Iterable[Dict[str, Any]]:
     else:
 
         def feature_gen() -> Iterable[Dict[str, Any]]:
-            yield json.loads(first_line)
+            def _cleanline(line: str):
+                # strip newlines
+                line = line.strip()
+                # strip spaces
+                line = line.strip()
+                if line.startswith('"'):
+                    line = line.strip('"')
+                return line
+            yield json.loads(_cleanline(first_line))
             for line in src:
-                yield json.loads(line)
+                yield json.loads(
+                    _cleanline(line)
+                )
 
     for obj in feature_gen():
         if isinstance(obj, list):