change: resolve git comments

Author: evakravi

tests/integ/sagemaker/jumpstart/script_mode_class/test_transfer_learning.py

@@ -31,7 +31,6 @@
     get_sm_session,
     get_training_dataset_for_model_and_version,
 )
-import pandas as pd
 
 
 def test_jumpstart_transfer_learning_estimator_class(setup):
@@ -123,10 +122,5 @@ def test_jumpstart_transfer_learning_estimator_class(setup):
     )
 
     response = endpoint_invoker.invoke_spc_endpoint(["hello", "world"])
-    entail, no_entail = response[0][0], response[0][1]
 
-    assert entail is not None
-    assert no_entail is not None
-
-    assert pd.isna(entail) is False
-    assert pd.isna(no_entail) is False
+    assert response is not None

tests/integ/sagemaker/jumpstart/utils.py

@@ -83,9 +83,9 @@ def download_file(local_download_path, s3_bucket, s3_key, s3_client) -> None:
 class EndpointInvoker:
     def __init__(
         self,
-        endpoint_name,
-        region=JUMPSTART_DEFAULT_REGION_NAME,
-        boto_config=Config(retries={"max_attempts": 10, "mode": "standard"}),
+        endpoint_name: str,
+        region: str = JUMPSTART_DEFAULT_REGION_NAME,
+        boto_config: Config = Config(retries={"max_attempts": 10, "mode": "standard"}),
     ) -> None:
         self.endpoint_name = endpoint_name
         self.region = region