Changes for unit tests in transformers build

Author: Bryannah Hernandez

src/sagemaker/serve/builder/transformers_builder.py

@@ -348,26 +348,28 @@ def _build_for_transformers(self):
         self.secret_key = None
         self.model_server = ModelServer.MMS
 
-        if not os.path.exists(self.model_path):
-            os.makedirs(self.model_path)
-
-        code_path = Path(self.model_path).joinpath("code")
-        # save the model or inference spec in cloud pickle format
         if self.inference_spec:
+
+            if not os.path.exists(self.model_path):
+                os.makedirs(self.model_path)
+
+     
+            code_path = Path(self.model_path).joinpath("code")
+
             save_pkl(code_path, (self.inference_spec, self.schema_builder))
             logger.info("PKL file saved to file: %s", code_path)
 
-        self._auto_detect_container()
+            self._auto_detect_container()
 
-        self.secret_key = prepare_for_mms(
-            model_path=self.model_path,
-            shared_libs=self.shared_libs,
-            dependencies=self.dependencies,
-            session=self.sagemaker_session,
-            image_uri=self.image_uri,
-            inference_spec=self.inference_spec,
-        )
+            self.secret_key = prepare_for_mms(
+                model_path=self.model_path,
+                    shared_libs=self.shared_libs,
+                    dependencies=self.dependencies,
+                    session=self.sagemaker_session,
+                    image_uri=self.image_uri,
+                    inference_spec=self.inference_spec,
+            )
 
-        self._build_transformers_env()
+        self._build_transformers_env()   
 
         return self.pysdk_model