Fix truss predict output. (#687)

basetenlabs · Sep 28, 2023 · 94b39f9 · 94b39f9
1 parent defd8f0
commit 94b39f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/truss/cli/cli.py b/truss/cli/cli.py
@@ -353,7 +353,7 @@ def predict(
     result = service.predict(request_data)
     if inspect.isgenerator(result):
         for chunk in result:
-            rich.print(chunk, end="")
+            click.echo(chunk, nl=False)
         return
     rich.print_json(data=result)