[fix] Propagate RollingBatch predict output code to request outputs (… (

#2495)

Co-authored-by: David Thomas <dthmasp@amazon.com>

engines/python/src/main/java/ai/djl/python/engine/RollingBatch.java

@@ -174,6 +174,11 @@ public void run() {
                     BytesSupplier err = BytesSupplier.wrap(JsonUtils.GSON.toJson(out));
                     for (Request req : list) {
                         req.last = true;
+
+                        // Note: This will only change the HTTP response code if the first chunk
+                        // has not yet been sent
+                        req.output.setCode(code);
+
                         req.data.appendContent(err, true);
                     }
                     list.clear();