return self after addition (deepjavalibrary#830)

KexinFeng · Aug 16, 2023 · 85fd906 · 85fd906
1 parent a279ae0
commit 85fd906
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/engines/python/setup/djl_python/outputs.py b/engines/python/setup/djl_python/outputs.py
@@ -135,6 +135,7 @@ def add_stream_content(self,
                            output_formatter=_default_stream_output_formatter):
         self.stream_content = stream_content
         self.stream_output_formatter = output_formatter
+        return self
 
     @staticmethod
     def _encode_json(val) -> bytes: