Minimal followup for BytessSupplier changes (#580)

This fixes serving for changes from deepjavalibrary/djl#2470. Some more extensive changes should follow including using the servingPredict and PublisherBytesSupplier. To fix CI, they will be part of a separate PR.
deepjavalibrary · Mar 29, 2023 · 7c63c23 · 7c63c23
1 parent 0874478
commit 7c63c23
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/engines/python/src/main/java/ai/djl/python/engine/Connection.java b/engines/python/src/main/java/ai/djl/python/engine/Connection.java
@@ -14,7 +14,7 @@
 
 import ai.djl.Model;
 import ai.djl.engine.EngineException;
-import ai.djl.modality.ChunkedBytesSupplier;
+import ai.djl.inference.streaming.ChunkedBytesSupplier;
 import ai.djl.modality.Input;
 import ai.djl.modality.Output;
 import ai.djl.ndarray.BytesSupplier;

diff --git a/engines/python/src/test/java/ai/djl/python/engine/PyEngineTest.java b/engines/python/src/test/java/ai/djl/python/engine/PyEngineTest.java
@@ -16,7 +16,7 @@
 import ai.djl.engine.Engine;
 import ai.djl.engine.EngineException;
 import ai.djl.inference.Predictor;
-import ai.djl.modality.ChunkedBytesSupplier;
+import ai.djl.inference.streaming.ChunkedBytesSupplier;
 import ai.djl.modality.Input;
 import ai.djl.modality.Output;
 import ai.djl.ndarray.BytesSupplier;

diff --git a/serving/src/main/java/ai/djl/serving/http/InferenceRequestHandler.java b/serving/src/main/java/ai/djl/serving/http/InferenceRequestHandler.java
@@ -13,8 +13,8 @@
 package ai.djl.serving.http;
 
 import ai.djl.ModelException;
+import ai.djl.inference.streaming.ChunkedBytesSupplier;
 import ai.djl.metric.Metric;
-import ai.djl.modality.ChunkedBytesSupplier;
 import ai.djl.modality.Input;
 import ai.djl.modality.Output;
 import ai.djl.ndarray.BytesSupplier;