Fix local import patch path

Author: samruds

tests/integ/sagemaker/serve/test_serve_pt_happy.py

@@ -10,6 +10,7 @@
 # distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
+# flake8: noqa: F631
 from __future__ import absolute_import
 
 import pytest
@@ -221,7 +222,7 @@ def test_happy_pytorch_sagemaker_endpoint(
             )
             if caught_ex:
                 logger.exception(caught_ex)
-                assert (  # noqa: F631
+                assert (
                     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,