rerun black and ai nit

sdk/ai/azure-ai-resources/azure/ai/resources/operations/_connection_operations.py

@@ -79,7 +79,7 @@ def get(self, name: str, scope: str = OperationScope.AI_RESOURCE, **kwargs) -> B
         """
         client = self._resource_ml_client if scope == OperationScope.AI_RESOURCE else self._project_ml_client
         workspace_connection = client._workspace_connections.get(name=name, **kwargs)
-        connection = BaseConnection._from_v2_workspace_connection(workspace_connection, params={"includeAll": "true"})
+        connection = BaseConnection._from_v2_workspace_connection(workspace_connection)
 
         # It's by design that both API and V2 SDK don't include the secrets from API response, the following
         # code fills the gap when possible

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_credentials.py

@@ -74,6 +74,7 @@
     UsernamePasswordAuthTypeWorkspaceConnectionProperties,
 )
 
+
 class _BaseIdentityConfiguration(ABC, DictMixin, RestTranslatableMixin):
     def __init__(self) -> None:
         self.type: Any = None
@@ -99,7 +100,9 @@ def _get_credential_class_from_rest_type(cls, auth_type: str) -> Type:
         }
         if not auth_type:
             return NoneCredentialConfiguration
-        return REST_CREDENTIAL_TYPE_TO_CLIENT_CLASS_MAP.get(_snake_to_camel(auth_type).lower(), NoneCredentialConfiguration)
+        return REST_CREDENTIAL_TYPE_TO_CLIENT_CLASS_MAP.get(
+            _snake_to_camel(auth_type).lower(), NoneCredentialConfiguration
+        )
 
 
 class AccountKeyConfiguration(RestTranslatableMixin, DictMixin):
@@ -162,7 +165,7 @@ def __eq__(self, other: object) -> bool:
 
     def __ne__(self, other: object) -> bool:
         return not self.__eq__(other)
-    
+
     @classmethod
     def _get_rest_properties_class(cls) -> Type:
         return SASAuthTypeWorkspaceConnectionProperties
@@ -207,6 +210,7 @@ def __eq__(self, other: object) -> bool:
     def _get_rest_properties_class(cls) -> Type:
         return PATAuthTypeWorkspaceConnectionProperties
 
+
 class UsernamePasswordConfiguration(RestTranslatableMixin, DictMixin):
     """Username and password credentials.
 
@@ -248,6 +252,7 @@ def __eq__(self, other: object) -> bool:
     def _get_rest_properties_class(cls) -> Type:
         return UsernamePasswordAuthTypeWorkspaceConnectionProperties
 
+
 class BaseTenantCredentials(RestTranslatableMixin, DictMixin, ABC):
     """Base class for tenant credentials.
 
@@ -354,6 +359,7 @@ def __ne__(self, other: object) -> bool:
     def _get_rest_properties_class(cls) -> Type:
         return ServicePrincipalAuthTypeWorkspaceConnectionProperties
 
+
 class CertificateConfiguration(BaseTenantCredentials):
     def __init__(
         self,
@@ -574,6 +580,7 @@ def __eq__(self, other: object) -> bool:
     def _get_rest_properties_class(cls) -> Type:
         return ManagedIdentityAuthTypeWorkspaceConnectionProperties
 
+
 class UserIdentityConfiguration(_BaseIdentityConfiguration):
     """User identity configuration.
 
@@ -816,6 +823,7 @@ def __ne__(self, other: object) -> bool:
     def _get_rest_properties_class(cls) -> Type:
         return NoneAuthTypeWorkspaceConnectionProperties
 
+
 class AccessKeyConfiguration(RestTranslatableMixin, DictMixin):
     """Access Key Credentials.
 
@@ -854,7 +862,7 @@ def __eq__(self, other: object) -> bool:
         if not isinstance(other, AccessKeyConfiguration):
             return NotImplemented
         return self.access_key_id == other.access_key_id and self.secret_access_key == other.secret_access_key
-    
+
     def _get_rest_properties_class(self):
         return AccessKeyAuthTypeWorkspaceConnectionProperties
 
@@ -893,6 +901,6 @@ def __eq__(self, other: object) -> bool:
         if not isinstance(other, ApiKeyConfiguration):
             return NotImplemented
         return bool(self.key == other.key)
-    
+
     def _get_rest_properties_class(self):
         return ApiKeyAuthWorkspaceConnectionProperties

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_workspace/connections/workspace_connection.py

@@ -307,11 +307,11 @@ def _to_rest_object(self) -> RestWorkspaceConnection:
         # No credential property bag uniquely has different inputs from ALL other propety bag classes.
         if workspace_connection_properties_class == NoneAuthTypeWorkspaceConnectionProperties:
             properties = workspace_connection_properties_class(
-                    target=self.target,
-                    metadata=self.tags,
-                    category=_snake_to_camel(conn_type),
-                    is_shared_to_all=self.is_shared,
-                )
+                target=self.target,
+                metadata=self.tags,
+                category=_snake_to_camel(conn_type),
+                is_shared_to_all=self.is_shared,
+            )
         else:
             properties = workspace_connection_properties_class(
                 target=self.target,

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_workspace/connections/workspace_connection_subtypes.py

@@ -293,7 +293,9 @@ def __init__(
         # Blob store connections returned from the API generally have no credentials, but we still don't want
         # to silently run over user inputted connections if they want to play with them locally, so double-check
         # kwargs for them.
-        super().__init__(target=target, type="azure_blob", credentials=kwargs.pop("credentials", None), from_child=True, **kwargs)
+        super().__init__(
+            target=target, type="azure_blob", credentials=kwargs.pop("credentials", None), from_child=True, **kwargs
+        )
 
         self.tags[CONNECTION_CONTAINER_NAME_KEY] = container_name
         self.tags[CONNECTION_ACCOUNT_NAME_KEY] = account_name

sdk/ml/azure-ai-ml/tests/workspace/unittests/test_workspace_connection_entity.py

@@ -10,6 +10,7 @@
 from azure.ai.ml.entities._credentials import PatTokenConfiguration
 from azure.ai.ml.constants._common import WorkspaceConnectionTypes
 
+
 @pytest.mark.unittest
 @pytest.mark.core_sdk_test
 class TestWorkspaceConnectionEntity: