Skip to content

Commit

Permalink
fix: validate kinesis extends to appends
Browse files Browse the repository at this point in the history
  • Loading branch information
RanbirAulakh committed Sep 24, 2024
1 parent b926e68 commit e641287
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion environment.yml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
name: osml_models
name: osml_model_runner_test
channels:
- conda-forge
dependencies:
Expand Down
4 changes: 2 additions & 2 deletions src/aws/osml/utils/integ_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -272,7 +272,7 @@ def validate_kinesis_features_match(
for record in records:
# Look for records that pertain to the target image_id
if record["PartitionKey"] == job_id:
kinesis_features.extend(geojson.loads(record["Data"])["features"])
kinesis_features.append(geojson.loads(record["Data"])["features"])
else:
logging.warning(f"Found partition key: {record['PartitionKey']}")
logging.warning(f"Looking for partition key: {job_id}")
Expand Down Expand Up @@ -402,9 +402,9 @@ def build_image_processing_request(endpoint: str, endpoint_type: str, image_url:
In the future this could, and probably should, be extended to build more variant image requests for additional
testing configurations.
:param endpoint: Model endpoint that you want to build the image_request for
:param endpoint_type: The type of endpoint you want to build the image_request for SM/HTTP
:param image_url: URL to the image you want to process
:param endpoint: Model endpoint that you want to build the image_request for
:return: Dictionary representation of the image request
"""
Expand Down

0 comments on commit e641287

Please sign in to comment.