def test_image(sagemaker_session): object2vec = Object2Vec(sagemaker_session=sagemaker_session, **ALL_REQ_ARGS) assert object2vec.train_image() == registry(REGION, "object2vec") + "/object2vec:1"