CI fixes

sdk/cosmos/azure-cosmos/azure/cosmos/_global_partition_endpoint_manager_per_partition_automatic_failover.py

@@ -20,7 +20,7 @@
 from azure.cosmos._routing.routing_range import PartitionKeyRangeWrapper
 
 if TYPE_CHECKING:
-    from azure.cosmos._cosmos_client_connection import CosmosClientConnection
+    from azure.cosmos.aio._cosmos_client_connection_async import CosmosClientConnection
 
 logger = logging.getLogger("azure.cosmos._GlobalPartitionEndpointManagerForPerPartitionAutomaticFailover")
 
@@ -126,8 +126,7 @@ def resolve_service_endpoint_for_partition(
                 partition_failover_info.current_regional_endpoint = request.location_endpoint_to_route
                 self.partition_range_to_failover_info[pk_range_wrapper] = partition_failover_info
             return self._resolve_service_endpoint(request)
-        else:
-            return self._resolve_service_endpoint_for_partition_circuit_breaker(request, pk_range_wrapper)
+        return self._resolve_service_endpoint_for_partition_circuit_breaker(request, pk_range_wrapper)
 
     def compute_available_preferred_regions(
             self,

sdk/cosmos/azure-cosmos/azure/cosmos/aio/_global_partition_endpoint_manager_per_partition_automatic_failover_async.py

@@ -58,7 +58,8 @@ def try_move_to_next_location(self, available_account_regional_endpoints: Set[st
 
             return False
 
-class _GlobalPartitionEndpointManagerForPerPartitionAutomaticFailoverAsync(_GlobalPartitionEndpointManagerForCircuitBreakerAsync):
+class _GlobalPartitionEndpointManagerForPerPartitionAutomaticFailoverAsync(
+    _GlobalPartitionEndpointManagerForCircuitBreakerAsync):
     """
     This internal class implements the logic for partition endpoint management for
     geo-replicated database accounts.
@@ -126,8 +127,7 @@ def resolve_service_endpoint_for_partition(
                 partition_failover_info.current_regional_endpoint = request.location_endpoint_to_route
                 self.partition_range_to_failover_info[pk_range_wrapper] = partition_failover_info
             return self._resolve_service_endpoint(request)
-        else:
-            return self._resolve_service_endpoint_for_partition_circuit_breaker(request, pk_range_wrapper)
+        return self._resolve_service_endpoint_for_partition_circuit_breaker(request, pk_range_wrapper)
 
     def compute_available_preferred_regions(
             self,

sdk/cosmos/azure-cosmos/tests/test_per_partition_automatic_failover.py

@@ -13,7 +13,7 @@
 from test_per_partition_circuit_breaker_mm import (REGION_1, REGION_2, PK_VALUE, BATCH,
                                                    write_operations_and_errors, perform_write_operation)
 
-# cspell:disable-file
+# cspell:disable
 
 @pytest.fixture(scope="class", autouse=True)
 def setup_teardown():

sdk/cosmos/azure-cosmos/tests/test_per_partition_automatic_failover_async.py

@@ -17,7 +17,7 @@
 from test_per_partition_circuit_breaker_mm import REGION_1, REGION_2, PK_VALUE, BATCH, write_operations_and_errors
 from test_per_partition_circuit_breaker_mm_async import perform_write_operation
 
-# cspell:disable-file
+# cspell:disable
 
 @pytest.fixture(scope="class", autouse=True)
 def setup_teardown():

sdk/cosmos/live-platform-matrix.json

@@ -61,7 +61,7 @@
     },
     {
       "PerPartitionAutomaticFailoverTestConfig": {
-        "Ubuntu2004_313_ppaf": {
+        "Ubuntu2004_313_partition_automatic_failover": {
           "OSVmImage": "env:LINUXVMIMAGE",
           "Pool": "env:LINUXPOOL",
           "PythonVersion": "3.13",