Manually overriding polling for delete (#40754)

* fixing delete functionality

Author: achauhan-scc

sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2025_01_01_preview/operations/_capability_hosts_operations.py

@@ -216,7 +216,7 @@ def _delete_initial(  # pylint: disable=inconsistent-return-statements
         response_headers = {}
         if response.status_code == 202:
             response_headers['x-ms-async-operation-timeout']=self._deserialize('duration', response.headers.get('x-ms-async-operation-timeout'))
-            response_headers['Location']=self._deserialize('str', response.headers.get('Location'))
+            response_headers['Azure-AsyncOperation']=self._deserialize('str', response.headers.get('Azure-AsyncOperation'))
             response_headers['Retry-After']=self._deserialize('int', response.headers.get('Retry-After'))
 
 
@@ -281,7 +281,7 @@ def get_long_running_output(pipeline_response):
                 return cls(pipeline_response, None, {})
 
 
-        if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'location'}, **kwargs)
+        if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'original-uri'}, **kwargs)
         elif polling is False: polling_method = NoPolling()
         else: polling_method = polling
         if cont_token: