bring back script assignment, cleanup test

Author: shreyapandit

src/sagemaker/model.py

@@ -476,7 +476,8 @@ def _script_mode_env_vars(self):
             else:
                 dir_name = self.uploaded_code.s3_prefix
         elif self.entry_point is not None:
-            if self.source_dir is not None:
+            script_name = self.entry_point
+        if self.source_dir is not None:
                 dir_name = (
                     self.source_dir
                     if self.source_dir.startswith("s3://")

tests/unit/sagemaker/model/test_framework_model.py

@@ -25,7 +25,6 @@
 MODEL_IMAGE = "mi"
 ENTRY_POINT = "blah.py"
 ROLE = "some-role"
-S3_SOURCE_DIR = "s3://somebucket/sourcedir.tar.gz"
 
 DATA_DIR = os.path.join(os.path.dirname(__file__), "..", "data")
 SCRIPT_NAME = "dummy_script.py"