address feedback

Author: ofek

tests/integ/sagemaker/conftest.py

@@ -280,7 +280,7 @@ def _generate_sagemaker_sdk_tar(destination_folder):
     """
     Run setup.py sdist to generate the PySDK tar file
     """
-    command = f"python3 setup.py egg_info --egg-base {destination_folder} sdist -d {destination_folder} -k --verbose"
+    command = f"python -m build --sdist -o {destination_folder}"
     print(f"Running command: {command}")
     result = subprocess.run(command, shell=True, check=True, capture_output=True)
     if result.returncode != 0:

tests/integ/sagemaker/feature_store/feature_processor/test_feature_processor.py

@@ -1272,7 +1272,7 @@ def _generate_and_move_sagemaker_sdk_tar():
     """
     Run setup.py sdist to generate the PySDK whl file
     """
-    subprocess.run("python3 setup.py bdist_wheel", shell=True)
+    subprocess.run("python -m build --wheel", shell=True)
     dist_dir = "dist"
     source_archive = os.listdir(dist_dir)[0]
     source_path = os.path.join(dist_dir, source_archive)