change pathways

neuralmagic · Dec 12, 2023 · fd5518e · fd5518e
1 parent a2ef0d1
commit fd5518e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/deepsparse/server/openai_server.py b/src/deepsparse/server/openai_server.py
@@ -19,6 +19,7 @@
 from typing import AsyncGenerator, Dict, List, Optional
 
 from deepsparse.legacy import Pipeline
+from deepsparse.legacy.tasks import SupportedTasks
 from deepsparse.server.config import EndpointConfig
 from deepsparse.server.helpers import create_error_response
 from deepsparse.server.output import CompletionOutput, RequestOutput
@@ -42,7 +43,6 @@
     random_uuid,
 )
 from deepsparse.server.server import Server
-from deepsparse.tasks import SupportedTasks
 from fastapi import BackgroundTasks, FastAPI, Request
 from fastapi.responses import StreamingResponse
 

diff --git a/tests/server/test_openai.py b/tests/server/test_openai.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import pytest
-from deepsparse import Pipeline
+from deepsparse.legacy import Pipeline
 from deepsparse.server.config import EndpointConfig, ServerConfig
 from deepsparse.server.openai_server import (
     ChatCompletionRequest,