Fix imports

src/cohere/manually_maintained/cohere_aws/chat.py

@@ -1,6 +1,6 @@
-from cohere_aws.response import CohereObject
-from cohere_aws.error import CohereError
-from cohere_aws.mode import Mode
+from .response import CohereObject
+from .error import CohereError
+from .mode import Mode
 from typing import List, Optional, Generator, Dict, Any, Union
 from enum import Enum
 import json

src/cohere/manually_maintained/cohere_aws/classification.py

@@ -1,4 +1,4 @@
-from cohere_aws.response import CohereObject
+from .response import CohereObject
 from typing import Any, Dict, Iterator, List, Literal, Union
 
 Prediction = Union[str, int, List[str], List[int]]

src/cohere/manually_maintained/cohere_aws/client.py

@@ -11,16 +11,16 @@
                                  ParamValidationError)
 from sagemaker.s3 import S3Downloader, S3Uploader, parse_s3_url
 
-from cohere_aws.classification import Classification, Classifications
-from cohere_aws.embeddings import Embeddings
-from cohere_aws.error import CohereError
-from cohere_aws.generation import (Generation, Generations,
+from .classification import Classification, Classifications
+from .embeddings import Embeddings
+from .error import CohereError
+from .generation import (Generation, Generations,
                                          StreamingGenerations,
                                          TokenLikelihood)
-from cohere_aws.chat import Chat, StreamingChat
-from cohere_aws.rerank import Reranking
-from cohere_aws.summary import Summary
-from cohere_aws.mode import Mode
+from .chat import Chat, StreamingChat
+from .rerank import Reranking
+from .summary import Summary
+from .mode import Mode
 
 
 class Client:

src/cohere/manually_maintained/cohere_aws/embeddings.py

@@ -1,4 +1,4 @@
-from cohere_aws.response import CohereObject
+from .response import CohereObject
 from typing import Iterator, List
 
 

src/cohere/manually_maintained/cohere_aws/generation.py

@@ -1,5 +1,5 @@
-from cohere_aws.response import CohereObject
-from cohere_aws.mode import Mode
+from .response import CohereObject
+from .mode import Mode
 from typing import List, Optional, NamedTuple, Generator, Dict, Any
 import json
 

src/cohere/manually_maintained/cohere_aws/rerank.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, Iterator, List, NamedTuple, Optional
 
-from cohere_aws.response import CohereObject
+from .response import CohereObject
 
 RerankDocument = NamedTuple("Document", [("text", str)])
 RerankDocument.__doc__ = """

src/cohere/manually_maintained/cohere_aws/summary.py

@@ -1,5 +1,5 @@
-from cohere_aws.error import CohereError
-from cohere_aws.response import CohereObject
+from .error import CohereError
+from .response import CohereObject
 from typing import Any, Dict, Optional