Fixing unit test failure

Author: sreedes

src/sagemaker/workflow/step_collections.py

@@ -108,6 +108,7 @@ def __init__(
         steps: List[Step] = []
         repack_model = False
         self.model_list = None
+        self.container_def_list = None
         if "entry_point" in kwargs:
             repack_model = True
             entry_point = kwargs.pop("entry_point", None)

tests/unit/sagemaker/workflow/test_step_collections.py

@@ -195,11 +195,7 @@ def test_register_model(estimator, model_metrics):
                 "Arguments": {
                     "InferenceSpecification": {
                         "Containers": [
-                            {
-                                "Image": "fakeimage",
-                                "ModelDataUrl": f"s3://{BUCKET}/model.tar.gz",
-                                "Environment": [],
-                            }
+                            {"Image": "fakeimage", "ModelDataUrl": f"s3://{BUCKET}/model.tar.gz"}
                         ],
                         "SupportedContentTypes": ["content_type"],
                         "SupportedRealtimeInferenceInstanceTypes": ["inference_instance"],
@@ -250,7 +246,6 @@ def test_register_model_tf(estimator_tf, model_metrics):
                             {
                                 "Image": "763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-inference:1.15.2-cpu",
                                 "ModelDataUrl": f"s3://{BUCKET}/model.tar.gz",
-                                "Environment": [],
                             }
                         ],
                         "SupportedContentTypes": ["content_type"],