Skip to content

Commit 4a6b3af

Browse files
bencrabtreebencrabtree
committed
update tests to pass
1 parent 3aad76b commit 4a6b3af

File tree

2 files changed

+57
-57
lines changed

2 files changed

+57
-57
lines changed

‎src/sagemaker/session.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6557,7 +6557,7 @@ def list_hubs(
65576557
if sort_order:
65586558
request["SortOrder"] = sort_order
65596559

6560-
return self.sagemaker_client.list_hubs(*request)
6560+
return self.sagemaker_client.list_hubs(**request)
65616561

65626562
def list_hub_contents(
65636563
self,
@@ -6607,7 +6607,7 @@ def list_hub_contents(
66076607
if sort_order:
66086608
request["SortOrder"] = sort_order
66096609

6610-
return self.sagemaker_client.list_hub_contents(*request)
6610+
return self.sagemaker_client.list_hub_contents(**request)
66116611

66126612
def delete_hub(
66136613
self,
@@ -6620,7 +6620,7 @@ def delete_hub(
66206620
"""
66216621
request = { "HubName": hub_name }
66226622

6623-
return self.sagemaker_client.delete_hub(*request)
6623+
return self.sagemaker_client.delete_hub(**request)
66246624

66256625
def import_hub_content(
66266626
self,
@@ -6675,7 +6675,7 @@ def import_hub_content(
66756675
if tags:
66766676
request["Tags"] = tags
66776677

6678-
return self.sagemaker_client.import_hub_content(*request)
6678+
return self.sagemaker_client.import_hub_content(**request)
66796679

66806680
def describe_hub_content(
66816681
self,
@@ -6703,7 +6703,7 @@ def describe_hub_content(
67036703
if hub_content_version:
67046704
request["HubContentVersion"] = hub_content_version
67056705

6706-
return self.sagemaker_client.describe_hub_content(*request)
6706+
return self.sagemaker_client.describe_hub_content(**request)
67076707

67086708
def delete_hub_content(
67096709
self,
@@ -6727,7 +6727,7 @@ def delete_hub_content(
67276727
"HubContentVersion": hub_content_version
67286728
}
67296729

6730-
return self.sagemaker_client.delete_hub_content(*request)
6730+
return self.sagemaker_client.delete_hub_content(**request)
67316731

67326732

67336733
def get_model_package_args(

‎tests/unit/test_session.py

Lines changed: 51 additions & 51 deletions
Original file line numberDiff line numberDiff line change
@@ -6503,33 +6503,33 @@ def test_create_hub(sagemaker_session):
65036503
)
65046504

65056505
request = {
6506-
"hub_name": "mock-hub-name",
6507-
"hub_description": "this is my sagemaker hub",
6508-
"hub_display_name": "Mock Hub",
6509-
"hub_search_keywords": ["mock", "hub", "123"],
6510-
"s3_storage_config": {
6506+
"HubName": "mock-hub-name",
6507+
"HubDescription": "this is my sagemaker hub",
6508+
"HubDisplayName": "Mock Hub",
6509+
"HubSearchKeywords": ["mock", "hub", "123"],
6510+
"S3StorageConfig": {
65116511
"S3OutputPath": "s3://my-hub-bucket/"
65126512
},
6513-
"tags": [{"Key": "tag-key-1", "Value": "tag-value-1"}]
6513+
"Tags": [{"Key": "tag-key-1", "Value": "tag-value-1"}]
65146514
}
65156515

6516-
sagemaker_session.create_hub.assert_called_with(**request)
6516+
sagemaker_session.sagemaker_client.create_hub.assert_called_with(**request)
65176517

65186518
def test_describe_hub(sagemaker_session):
65196519
sagemaker_session.describe_hub(
65206520
hub_name="mock-hub-name",
65216521
)
65226522

65236523
request = {
6524-
"hub_name": "mock-hub-name",
6524+
"HubName": "mock-hub-name",
65256525
}
65266526

6527-
sagemaker_session.describe_hub.assert_called_with(**request)
6527+
sagemaker_session.sagemaker_client.describe_hub.assert_called_with(**request)
65286528

65296529
def test_list_hubs(sagemaker_session):
65306530
sagemaker_session.list_hubs(
65316531
creation_time_after="08-14-1997 12:00:00",
6532-
creation_time_before="01-08/2024 10:25:00",
6532+
creation_time_before="01-08-2024 10:25:00",
65336533
max_results="25",
65346534
max_schema_version="1.0.5",
65356535
name_contains="mock-hub",
@@ -6538,16 +6538,16 @@ def test_list_hubs(sagemaker_session):
65386538
)
65396539

65406540
request = {
6541-
"creation_time_after": "08-14-1997 12:00:00",
6542-
"creation_time_before": "01-08/2024 10:25:00",
6543-
"max_results": "25",
6544-
"max_schema_version": "1.0.5",
6545-
"name_contains": "mock-hub",
6546-
"sort_by": "HubName",
6547-
"sort_order": "Ascending"
6541+
"CreationTimeAfter": "08-14-1997 12:00:00",
6542+
"CreationTimeBefore": "01-08-2024 10:25:00",
6543+
"MaxResults": "25",
6544+
"MaxSchemaVersion": "1.0.5",
6545+
"NameContains": "mock-hub",
6546+
"SortBy": "HubName",
6547+
"SortOrder": "Ascending"
65486548
}
65496549

6550-
sagemaker_session.list_hubs.assert_called_with(**request)
6550+
sagemaker_session.sagemaker_client.list_hubs.assert_called_with(**request)
65516551

65526552
def test_list_hub_contents(sagemaker_session):
65536553
sagemaker_session.list_hub_contents(
@@ -6563,29 +6563,29 @@ def test_list_hub_contents(sagemaker_session):
65636563
)
65646564

65656565
request = {
6566-
"hub_name": "mock-hub-123",
6567-
"hub_content_type": "MODEL",
6568-
"creation_time_after": "08-14-1997 12:00:00",
6569-
"creation_time_before": "01-08/2024 10:25:00",
6570-
"max_results": "25",
6571-
"max_schema_version": "1.0.5",
6572-
"name_contains": "mock-hub",
6573-
"sort_by": "HubName",
6574-
"sort_order": "Ascending"
6566+
"HubName": "mock-hub-123",
6567+
"HubContentType": "MODEL",
6568+
"CreationTimeAfter": "08-14-1997 12:00:00",
6569+
"CreationTimeBefore": "01-08/2024 10:25:00",
6570+
"MaxResults": "25",
6571+
"MaxSchemaVersion": "1.0.5",
6572+
"NameContains": "mock-hub",
6573+
"SortBy": "HubName",
6574+
"SortOrder": "Ascending"
65756575
}
65766576

6577-
sagemaker_session.list_hub_contents.assert_called_with(**request)
6577+
sagemaker_session.sagemaker_client.list_hub_contents.assert_called_with(**request)
65786578

65796579
def test_delete_hub(sagemaker_session):
65806580
sagemaker_session.delete_hub(
65816581
hub_name="mock-hub-123",
65826582
)
65836583

65846584
request = {
6585-
"hub_name": "mock-hub-123",
6585+
"HubName": "mock-hub-123",
65866586
}
65876587

6588-
sagemaker_session.delete_hub.assert_called_with(**request)
6588+
sagemaker_session.sagemaker_client.delete_hub.assert_called_with(**request)
65896589

65906590
def test_import_hub_content(sagemaker_session):
65916591
sagemaker_session.import_hub_content(
@@ -6602,19 +6602,19 @@ def test_import_hub_content(sagemaker_session):
66026602
)
66036603

66046604
request = {
6605-
"hub_name": "mock-hub-123",
6606-
"hub_content_type": "MODEL",
6607-
"document_schema_version": "1.0.0",
6608-
"hub_content_document": "{'training_script_location':'s3://path/to/script.py'}",
6609-
"hub_content_name": "mock-hub-content-1",
6610-
"hub_content_display_name": "Mock Hub Content One",
6611-
"hub_content_description": "This is my special Hub Content for my special Hub",
6612-
"hub_content_version": "5.5.5",
6613-
"hub_content_markdown": "markdown",
6614-
"hub_content_search_keywords": ["Hub","Machine Learning","Content"]
6605+
"HubName": "mock-hub-123",
6606+
"HubContentType": "MODEL",
6607+
"DocumentSchemaVersion": "1.0.0",
6608+
"HubContentDocument": "{'training_script_location':'s3://path/to/script.py'}",
6609+
"HubContentName": "mock-hub-content-1",
6610+
"HubContentDisplayName": "Mock Hub Content One",
6611+
"HubContentDescription": "This is my special Hub Content for my special Hub",
6612+
"HubContentVersion": "5.5.5",
6613+
"HubContentMarkdown": "markdown",
6614+
"HubContentSearchKeywords": ["Hub","Machine Learning","Content"]
66156615
}
66166616

6617-
sagemaker_session.import_hub_content.assert_called_with(**request)
6617+
sagemaker_session.sagemaker_client.import_hub_content.assert_called_with(**request)
66186618

66196619
def test_describe_hub_content(sagemaker_session):
66206620
sagemaker_session.describe_hub_content(
@@ -6625,13 +6625,13 @@ def test_describe_hub_content(sagemaker_session):
66256625
)
66266626

66276627
request = {
6628-
"hub_name": "mock-hub-123",
6629-
"hub_content_type": "MODEL",
6630-
"hub_content_name": "mock-hub-content-1",
6631-
"hub_content_version": "5.5.5",
6628+
"HubName": "mock-hub-123",
6629+
"HubContentType": "MODEL",
6630+
"HubContentName": "mock-hub-content-1",
6631+
"HubContentVersion": "5.5.5",
66326632
}
66336633

6634-
sagemaker_session.describe_hub_content.assert_called_with(**request)
6634+
sagemaker_session.sagemaker_client.describe_hub_content.assert_called_with(**request)
66356635

66366636
def test_delete_hub_content(sagemaker_session):
66376637
sagemaker_session.delete_hub_content(
@@ -6642,10 +6642,10 @@ def test_delete_hub_content(sagemaker_session):
66426642
)
66436643

66446644
request = {
6645-
"hub_name": "mock-hub-123",
6646-
"hub_content_type": "MODEL",
6647-
"hub_content_name": "mock-hub-content-1",
6648-
"hub_content_version": "5.5.5",
6645+
"HubName": "mock-hub-123",
6646+
"HubContentType": "MODEL",
6647+
"HubContentName": "mock-hub-content-1",
6648+
"HubContentVersion": "5.5.5",
66496649
}
66506650

6651-
sagemaker_session.delete_hub_content.assert_called_with(**request)
6651+
sagemaker_session.sagemaker_client.delete_hub_content.assert_called_with(**request)

0 commit comments

Comments
 (0)