Fix local import patch path

Author: samruds

tests/integ/sagemaker/serve/test_serve_pt_happy.py

@@ -221,7 +221,7 @@ def test_happy_pytorch_sagemaker_endpoint(
             )
             if caught_ex:
                 logger.exception(caught_ex)
-                assert (  # noqa: F631
+                assert (  # flake8: noqa: F631
                     False,
                 ), f"{caught_ex} was thrown when running pytorch squeezenet sagemaker endpoint test"
 

tests/unit/sagemaker/serve/utils/test_hardware_detector.py

@@ -101,8 +101,8 @@ def test_format_instance_type_without_ml_success():
     assert formatted_instance_type == "g5.48xlarge"
 
 
-@patch("sagemaker.serve.utils.hardware_detector.estimate_command_parser")
-@patch("sagemaker.serve.utils.hardware_detector.gather_data")
+@patch("accelerate.commands.estimate.estimate_command_parser")
+@patch("accelerate.commands.estimate.gather_data")
 def test_total_inference_model_size_mib(
     mock_gather_data,
     mock_parser,