test fix

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

tests/unit/test_telemetry.py

@@ -91,7 +91,7 @@ def test_network_request_flow(self, mock_post, mock_telemetry_client):
         args, kwargs = client._executor.submit.call_args
 
         # Verify correct function and URL
-        assert args[0] == requests.post
+        assert args[0] == client._session.post
         assert args[1] == 'https://test-host.com/telemetry-ext'
         assert kwargs['headers']['Authorization'] == 'Bearer test-token'