diff --git a/weave/legacy/io_service.py b/weave/legacy/io_service.py index 50502c700e4..a5e0c37f554 100644 --- a/weave/legacy/io_service.py +++ b/weave/legacy/io_service.py @@ -26,10 +26,10 @@ engine_trace, errors, filesystem, - server_error_handling, weave_http, ) from weave.legacy import ( + server_error_handling, artifact_wandb, async_queue, cache, diff --git a/weave/server_error_handling.py b/weave/legacy/server_error_handling.py similarity index 99% rename from weave/server_error_handling.py rename to weave/legacy/server_error_handling.py index 35f20ab7852..bdc5a46ee2b 100644 --- a/weave/server_error_handling.py +++ b/weave/legacy/server_error_handling.py @@ -27,7 +27,7 @@ from werkzeug import exceptions as werkzeug_exceptions from werkzeug import http as werkzeug_http -from . import errors, util +from .. import errors, util class WeaveInternalHttpException(werkzeug_exceptions.HTTPException): diff --git a/weave/weave_http.py b/weave/weave_http.py index 43e1d96116b..8d35a00b5c0 100644 --- a/weave/weave_http.py +++ b/weave/weave_http.py @@ -13,7 +13,7 @@ import requests.auth import yarl -from weave import server_error_handling +from weave.legacy import server_error_handling from . import engine_trace, filesystem diff --git a/weave/weave_server.py b/weave/weave_server.py index 461fe55989d..4ec00b16358 100644 --- a/weave/weave_server.py +++ b/weave/weave_server.py @@ -44,7 +44,7 @@ weavejs_fixes, ) from weave.legacy.language_features.tagging import tag_store -from weave.server_error_handling import client_safe_http_exceptions_as_werkzeug +from weave.legacy.server_error_handling import client_safe_http_exceptions_as_werkzeug logger = logging.getLogger(__name__)