Skip to content

Commit

Permalink
linting fix
Browse files Browse the repository at this point in the history
  • Loading branch information
liustve committed Dec 12, 2024
1 parent 6fe1ca7 commit fcffa19
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -429,14 +429,14 @@ def get_model_request_response(path):

if "amazon.nova" in path:
model_id = "amazon.nova-pro-v1:0"

request_body = {
"messages":[{"role": "user", "content": [{"text": "A camping trip"}]}],
"messages": [{"role": "user", "content": [{"text": "A camping trip"}]}],
"inferenceConfig": {
"max_new_tokens": 800,
"temperature": 0.9,
"top_p": 0.7,
}
},
}

response_body = {
Expand Down
6 changes: 4 additions & 2 deletions contract-tests/tests/test/amazon/botocore/botocore_test.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0
import math
from logging import INFO, Logger, getLogger
import re
from logging import INFO, Logger, getLogger
from typing import Dict, List

from docker.types import EndpointConfig
Expand Down Expand Up @@ -1074,4 +1074,6 @@ def _assert_str_attribute(self, attributes_dict: Dict[str, AnyValue], key: str,
self.assertIsNotNone(actual_value)
pattern = re.compile(expected_value)
match = pattern.fullmatch(actual_value.string_value)
self.assertTrue(match is not None, f"Actual: {actual_value.string_value} does not match Expected: {expected_value}")
self.assertTrue(
match is not None, f"Actual: {actual_value.string_value} does not match Expected: {expected_value}"
)
Original file line number Diff line number Diff line change
Expand Up @@ -18,4 +18,4 @@ def server_request():


if __name__ == "__main__":
app.run(port=8082)
app.run(port=8082)

0 comments on commit fcffa19

Please sign in to comment.