fix: failing integ test

evakravi · evakravi · commit 64ca588e4a32 · 2023-05-05T17:02:50.000Z
diff --git a/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py b/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py
@@ -66,7 +66,7 @@ def test_prepacked_jumpstart_model(setup):
         tags=[{"Key": JUMPSTART_TAG, "Value": os.environ[ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID]}],
     )
 
-    response = predictor.predict({"prompt": "hello world!"})
+    response = predictor.predict("hello world!")
 
     assert response is not None
 

Original file line number	Diff line number	Diff line change
`@@ -66,7 +66,7 @@ def test_prepacked_jumpstart_model(setup):`
`66`	`66`	`tags=[{"Key": JUMPSTART_TAG, "Value": os.environ[ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID]}],`
`67`	`67`	`)`
`68`	`68`
`69`		`- response = predictor.predict({"prompt": "hello world!"})`
	`69`	`+ response = predictor.predict("hello world!")`
`70`	`70`
`71`	`71`	`assert response is not None`
`72`	`72`