fix: do not delete temp folder generated by sdist (#1291)

Co-authored-by: Rohan Gujarathi <[email protected]>

Author: 2 people

tests/integ/sagemaker/conftest.py

@@ -248,7 +248,11 @@ def _generate_sagemaker_sdk_tar(destination_folder):
     """
     Run setup.py sdist to generate the PySDK tar file
     """
-    subprocess.run(f"python3 setup.py sdist -d {destination_folder}", shell=True)
+    subprocess.run(
+        f"python3 setup.py egg_info --egg-base {destination_folder} sdist -d {destination_folder} -k",
+        shell=True,
+        check=True,
+    )
     destination_folder_contents = os.listdir(destination_folder)
     source_archive = [file for file in destination_folder_contents if file.endswith("tar.gz")][0]
 

tests/integ/sagemaker/experiments/conftest.py

@@ -155,7 +155,7 @@ def tempdir():
 @pytest.fixture(scope="module")
 def dev_sdk_tar():
     resource_dir = os.path.join(DATA_DIR, "experiment")
-    os.system("python setup.py sdist")
+    os.system("python setup.py sdist -k")
     sdist_path = max(glob.glob("dist/sagemaker-*"), key=os.path.getctime)
     sdk_file = os.path.join(resource_dir, _EXP_PLUS_SDK_TAR)
     shutil.copy(sdist_path, sdk_file)