Update api_test.py

Author: yikaiMeta

torchx/runner/test/api_test.py

@@ -105,7 +105,7 @@ def test_validate_invalid_replicas(self, _) -> None:
             with self.assertRaises(ValueError):
                 runner.run(app, scheduler="local_dir")
 
-    @patch("torchx.util.session.uuid.uuid4")
+    @patch("uuid.uuid4")
     def test_session_id(self, uuid_mock: MagicMock, record_mock: MagicMock) -> None:
         uuid_mock.return_value = "test_session_id"
         test_file = self.tmpdir / "test_file"
@@ -134,7 +134,7 @@ def test_session_id(self, uuid_mock: MagicMock, record_mock: MagicMock) -> None:
                 event = record_mock.call_args_list[i].args[0]
                 self.assertEqual(event.session, "test_session_id")
 
-    @patch("torchx.util.session.uuid.uuid4")
+    @patch("uuid.uuid4")
     def test_run(self, uuid_mock: MagicMock, _) -> None:
         uuid_mock.return_value = "test_session_id"
         test_file = self.tmpdir / "test_file"