Skip to content

Commit a043f0c

Browse files
jscudcopybara-github
authored andcommitted
chore: Debugging flaky test.
PiperOrigin-RevId: 931321273
1 parent b51f0f6 commit a043f0c

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

tests/system/aiplatform/test_experiment_model.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -363,5 +363,14 @@ def test_deploy_model_with_gpu_container(self, shared_state):
363363
)
364364

365365
pred = endpoint.predict([[1, 2]])
366+
367+
print("prediction: ", pred)
368+
print("prediction type: ", type(pred))
369+
with open("/tmp/pred1", "w") as f:
370+
f.write(str(pred))
371+
f.write("\n\n********\n\n\n")
372+
f.write(str(type(pred)))
373+
f.close()
374+
366375
assert isinstance(pred, models.Prediction)
367376
shared_state["resources"].append(endpoint)

0 commit comments

Comments
 (0)