fix imports to remove deprecation warning

Signed-off-by: Daniele Trifirò <[email protected]>
caikit · Sep 27, 2023 · 53581ee · 53581ee
1 parent 9342cc4
commit 53581ee
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/caikit_tgis_backend/managed_tgis_subprocess.py b/caikit_tgis_backend/managed_tgis_subprocess.py
@@ -29,7 +29,7 @@
 import requests
 
 # First Party
-from caikit.core.toolkit.errors import error_handler
+from caikit.core.exceptions import error_handler
 import alog
 
 # Local

diff --git a/caikit_tgis_backend/tgis_backend.py b/caikit_tgis_backend/tgis_backend.py
@@ -22,9 +22,9 @@
 import grpc
 
 # First Party
+from caikit.core.exceptions import error_handler
 from caikit.core.module_backends.backend_types import register_backend_type
 from caikit.core.module_backends.base import BackendBase
-from caikit.core.toolkit.errors import error_handler
 import alog
 
 # Local

diff --git a/caikit_tgis_backend/tgis_connection.py b/caikit_tgis_backend/tgis_connection.py
@@ -23,7 +23,7 @@
 import grpc
 
 # First Party
-from caikit.core.toolkit.errors import error_handler
+from caikit.core.exceptions import error_handler
 import alog
 
 # Local