test: fix imports test

AI21Labs · Jun 27, 2024 · e0cc96a · e0cc96a
1 parent 26fe85b
commit e0cc96a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/unittests/test_imports.py b/tests/unittests/test_imports.py
@@ -20,6 +20,7 @@
     "SageMaker",
     "TooManyRequestsError",
     "AsyncAI21BedrockClient",
+    "AsyncAI21SageMakerClient",
 ]