def scale_down_a3_cluster()

in dags/multipod/maxtext_gpu_end_to_end.py [0:0]

• 23 lines of code
• 1 McCabe index (conditional complexity)

def scale_down_a3_cluster():
  with tempfile.TemporaryDirectory() as tmpdir:
    hook = SubprocessHook()

    result = hook.run_command(
        [
            "bash",
            "-c",
            ";".join(
                configure_project_and_cluster(
                    Project.SUPERCOMPUTER_TESTING.value,
                    XpkClusters.GPU_A3_CLUSTER.name,
                    XpkClusters.GPU_A3_CLUSTER.zone,
                )
                + resize_a3_cluster(
                    XpkClusters.GPU_A3_CLUSTER.name,
                    XpkClusters.GPU_A3_CLUSTER.zone,
                    0,
                )
            ),
        ],
        cwd=tmpdir,
    )
    assert result.exit_code == 0, f"Command failed with code {result.exit_code}"