Fix black formatting issues in cloud provider files and notebook magic

- Auto-formatted clustrix/cloud_providers/gcp.py
- Auto-formatted clustrix/cloud_providers/azure.py
- Auto-formatted clustrix/cloud_providers/aws.py
- Auto-formatted clustrix/notebook_magic.py

All linting checks now pass and tests remain functional.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: jeremymanning

clustrix/cloud_providers/aws.py

@@ -473,13 +473,14 @@ def delete_cluster(
                     for nodegroup_name in nodegroups.get("nodegroups", []):
                         logger.info(f"Deleting node group: {nodegroup_name}")
                         self.eks_client.delete_nodegroup(
-                            clusterName=cluster_identifier,
-                            nodegroupName=nodegroup_name
+                            clusterName=cluster_identifier, nodegroupName=nodegroup_name
                         )
 
                     # Delete the cluster itself
                     self.eks_client.delete_cluster(name=cluster_identifier)
-                    logger.info(f"EKS cluster '{cluster_identifier}' deletion initiated")
+                    logger.info(
+                        f"EKS cluster '{cluster_identifier}' deletion initiated"
+                    )
                     return True
 
                 except ClientError as e:
@@ -524,9 +525,13 @@ def get_cluster_status(
                         # Get details of first node group for node count
                         ng_response = self.eks_client.describe_nodegroup(
                             clusterName=cluster_identifier,
-                            nodegroupName=nodegroups["nodegroups"][0]
+                            nodegroupName=nodegroups["nodegroups"][0],
+                        )
+                        node_count = (
+                            ng_response["nodegroup"]
+                            .get("scalingConfig", {})
+                            .get("desiredSize", 0)
                         )
-                        node_count = ng_response["nodegroup"].get("scalingConfig", {}).get("desiredSize", 0)
 
                     return {
                         "cluster_name": cluster_identifier,

clustrix/cloud_providers/azure.py

@@ -456,9 +456,11 @@ def delete_cluster(self, cluster_identifier: str, cluster_type: str = "vm") -> b
                 # Delete AKS cluster
                 operation = self.container_client.managed_clusters.begin_delete(
                     resource_group_name=self.resource_group,
-                    resource_name=cluster_identifier
+                    resource_name=cluster_identifier,
+                )
+                logger.info(
+                    f"Deleting AKS cluster '{cluster_identifier}' - operation: {operation}"
                 )
-                logger.info(f"Deleting AKS cluster '{cluster_identifier}' - operation: {operation}")
                 return True
             else:
                 raise ValueError(f"Unknown cluster type: {cluster_type}")
@@ -500,7 +502,7 @@ def get_cluster_status(
                 # Get AKS cluster status
                 cluster = self.container_client.managed_clusters.get(
                     resource_group_name=self.resource_group,
-                    resource_name=cluster_identifier
+                    resource_name=cluster_identifier,
                 )
 
                 return {
@@ -563,8 +565,10 @@ def list_clusters(self) -> List[Dict[str, Any]]:
 
         # List AKS clusters
         try:
-            aks_clusters = self.container_client.managed_clusters.list_by_resource_group(
-                resource_group_name=self.resource_group
+            aks_clusters = (
+                self.container_client.managed_clusters.list_by_resource_group(
+                    resource_group_name=self.resource_group
+                )
             )
 
             for cluster in aks_clusters:

clustrix/cloud_providers/gcp.py

@@ -274,16 +274,10 @@ def create_gke_cluster(
                     },
                 },
                 "master_auth": {
-                    "client_certificate_config": {
-                        "issue_client_certificate": False
-                    }
-                },
-                "ip_allocation_policy": {
-                    "use_ip_aliases": True
-                },
-                "network_policy": {
-                    "enabled": False
+                    "client_certificate_config": {"issue_client_certificate": False}
                 },
+                "ip_allocation_policy": {"use_ip_aliases": True},
+                "network_policy": {"enabled": False},
                 "addons_config": {
                     "http_load_balancing": {"disabled": False},
                     "horizontal_pod_autoscaling": {"disabled": False},
@@ -361,7 +355,9 @@ def delete_cluster(
                 request = container_v1.DeleteClusterRequest(name=name)
                 operation = self.container_client.delete_cluster(request=request)
 
-                logger.info(f"Deleting GKE cluster '{cluster_identifier}' - operation: {operation.name}")
+                logger.info(
+                    f"Deleting GKE cluster '{cluster_identifier}' - operation: {operation.name}"
+                )
                 return True
             else:
                 raise ValueError(f"Unknown cluster type: {cluster_type}")

clustrix/notebook_magic.py

@@ -480,8 +480,10 @@ def _create_widgets(self):
                 "huggingface",
             ],
             description="Cluster Type:",
-            tooltip=("Choose where to run your jobs: local machine, remote servers "
-                     "(SSH/SLURM/PBS/SGE), Kubernetes clusters, or cloud providers"),
+            tooltip=(
+                "Choose where to run your jobs: local machine, remote servers "
+                "(SSH/SLURM/PBS/SGE), Kubernetes clusters, or cloud providers"
+            ),
             style=style,
             layout=full_layout,
         )
@@ -492,8 +494,10 @@ def _create_widgets(self):
         self.config_name = widgets.Text(
             description="Config Name:",
             placeholder="Enter configuration name",
-            tooltip=("Give this configuration a descriptive name "
-                     "(e.g., 'AWS Production', 'Local Testing', 'HPC Cluster')"),
+            tooltip=(
+                "Give this configuration a descriptive name "
+                "(e.g., 'AWS Production', 'Local Testing', 'HPC Cluster')"
+            ),
             style=style,
             layout=full_layout,
         )
@@ -586,8 +590,10 @@ def _create_dynamic_fields(self):
         self.host_field = widgets.Text(
             description="Host/Address:",
             placeholder="hostname or IP address",
-            tooltip=("Enter the hostname or IP address of your remote cluster "
-                     "(e.g., cluster.example.com or 192.168.1.100)"),
+            tooltip=(
+                "Enter the hostname or IP address of your remote cluster "
+                "(e.g., cluster.example.com or 192.168.1.100)"
+            ),
             style=style,
             layout=full_layout,
         )
@@ -604,8 +610,10 @@ def _create_dynamic_fields(self):
         self.ssh_key_field = widgets.Text(
             description="SSH Key:",
             placeholder="~/.ssh/id_rsa",
-            tooltip=("Path to your SSH private key file for passwordless authentication "
-                     "(generate with 'ssh-keygen -t rsa')"),
+            tooltip=(
+                "Path to your SSH private key file for passwordless authentication "
+                "(generate with 'ssh-keygen -t rsa')"
+            ),
             style=style,
             layout=half_layout,
         )
@@ -629,8 +637,10 @@ def _create_dynamic_fields(self):
             value="8GB",
             description="Memory:",
             placeholder="e.g., 8GB, 16GB",
-            tooltip=("Amount of RAM to request for each job (e.g., '8GB', '16GB'). "
-                     "Higher memory allows processing larger datasets"),
+            tooltip=(
+                "Amount of RAM to request for each job (e.g., '8GB', '16GB'). "
+                "Higher memory allows processing larger datasets"
+            ),
             style=style,
             layout=widgets.Layout(width="200px"),
         )
@@ -646,8 +656,10 @@ def _create_dynamic_fields(self):
         self.k8s_namespace = widgets.Text(
             value="default",
             description="Namespace:",
-            tooltip=("Kubernetes namespace for job pods (default: 'default'). "
-                     "Contact your cluster admin for appropriate namespace"),
+            tooltip=(
+                "Kubernetes namespace for job pods (default: 'default'). "
+                "Contact your cluster admin for appropriate namespace"
+            ),
             style=style,
             layout=half_layout,
         )
@@ -756,16 +768,20 @@ def _create_dynamic_fields(self):
         self.azure_client_id = widgets.Text(
             description="Client ID:",
             placeholder="Azure service principal client ID",
-            tooltip=("Azure service principal application ID (UUID format). "
-                     "Create in Azure AD > App registrations"),
+            tooltip=(
+                "Azure service principal application ID (UUID format). "
+                "Create in Azure AD > App registrations"
+            ),
             style=style,
             layout=half_layout,
         )
         self.azure_client_secret = widgets.Password(
             description="Client Secret:",
             placeholder="Azure service principal secret",
-            tooltip=("Azure service principal secret (keep secure!). Generated in Azure AD > "
-                     "App registrations > Certificates & secrets"),
+            tooltip=(
+                "Azure service principal secret (keep secure!). Generated in Azure AD > "
+                "App registrations > Certificates & secrets"
+            ),
             style=style,
             layout=half_layout,
         )
@@ -782,8 +798,10 @@ def _create_dynamic_fields(self):
             options=["us-central1"],  # Will be populated dynamically
             value="us-central1",
             description="GCP Region:",
-            tooltip=("Google Cloud region for resources (affects latency and pricing). "
-                     "Choose region closest to your location"),
+            tooltip=(
+                "Google Cloud region for resources (affects latency and pricing). "
+                "Choose region closest to your location"
+            ),
             style=style,
             layout=half_layout,
         )
@@ -800,8 +818,10 @@ def _create_dynamic_fields(self):
         self.gcp_service_account_key = widgets.Textarea(
             description="Service Account Key:",
             placeholder="Paste your GCP service account JSON key here",
-            tooltip=("Google Cloud service account key in JSON format (keep secure!). "
-                     "Create in GCP Console > IAM & Admin > Service Accounts"),
+            tooltip=(
+                "Google Cloud service account key in JSON format (keep secure!). "
+                "Create in GCP Console > IAM & Admin > Service Accounts"
+            ),
             style=style,
             layout=full_layout,
         )
@@ -818,8 +838,10 @@ def _create_dynamic_fields(self):
             options=["gpu_1x_a10"],  # Will be populated dynamically
             value="gpu_1x_a10",
             description="Instance Type:",
-            tooltip=("Lambda Cloud GPU instance type (affects GPU model, RAM, and cost). "
-                     "gpu_1x_a10 = 1x NVIDIA A10 + 30GB RAM"),
+            tooltip=(
+                "Lambda Cloud GPU instance type (affects GPU model, RAM, and cost). "
+                "gpu_1x_a10 = 1x NVIDIA A10 + 30GB RAM"
+            ),
             style=style,
             layout=half_layout,
         )
@@ -828,8 +850,10 @@ def _create_dynamic_fields(self):
         self.hf_token = widgets.Password(
             description="HF Token:",
             placeholder="Your HuggingFace API token",
-            tooltip=("HuggingFace API token for Spaces access (keep secure!). "
-                     "Get from HuggingFace Settings > Access Tokens"),
+            tooltip=(
+                "HuggingFace API token for Spaces access (keep secure!). "
+                "Get from HuggingFace Settings > Access Tokens"
+            ),
             style=style,
             layout=half_layout,
         )