test: check imports

Author: etang-ai21

ai21/clients/sagemaker/resources/sagemaker_completion.py

@@ -1,4 +1,4 @@
-from __future__ import annotations
+# from __future__ import annotations
 
 from typing import List, Dict
 

tests/unittests/test_imports.py

@@ -0,0 +1,21 @@
+from ai21 import __all__
+from ai21 import *  # noqa: F403
+
+EXPECTED_ALL = [
+    "AI21EnvConfig",
+    "AI21Client",
+    "AI21APIError",
+    "APITimeoutError",
+    "AI21Error",
+    "MissingApiKeyError",
+    "ModelPackageDoesntExistError",
+    "TooManyRequestsError",
+    "AI21BedrockClient",
+    "AI21SageMakerClient",
+    "BedrockModelID",
+    "SageMaker",
+]
+
+
+def test_all_imports() -> None:
+    assert sorted(EXPECTED_ALL) == sorted(__all__)