flywheel-bootstrap-staging 0.1.9.202602012332__tar.gz → 0.1.9.202602020926__tar.gz

{flywheel_bootstrap_staging-0.1.9.202602012332 → flywheel_bootstrap_staging-0.1.9.202602020926}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flywheel-bootstrap-staging
-Version: 0.1.9.202602012332
+Version: 0.1.9.202602020926
 Summary: Bootstrap runner for Flywheel provisioned GPU instances
 Project-URL: Homepage, http://paradigma.inc/
 Author: Paradigma Labs

{flywheel_bootstrap_staging-0.1.9.202602012332 → flywheel_bootstrap_staging-0.1.9.202602020926}/bootstrap/git_ops.py

@@ -200,10 +200,20 @@ def setup_branch(config: GitConfig) -> bool:
     )
 
     if result.returncode == 0 and branch_name in result.stdout:
-        # Branch exists, check it out
+        # Branch exists, fetch it explicitly (single-branch clones don't fetch other heads)
         config.log("info", f"Checking out existing branch: {branch_name}")
+        fetch_result = _run_git(
+            ["fetch", "origin", branch_name],
+            cwd=workspace,
+        )
+        if fetch_result.returncode != 0:
+            config.log(
+                "error", f"Failed to fetch branch {branch_name}: {fetch_result.stderr}"
+            )
+            return False
+
         result = _run_git(
-            ["checkout", "-B", branch_name, f"origin/{branch_name}"],
+            ["checkout", "-B", branch_name, "FETCH_HEAD"],
             cwd=workspace,
         )
     else:

{flywheel_bootstrap_staging-0.1.9.202602012332 → flywheel_bootstrap_staging-0.1.9.202602020926}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "flywheel-bootstrap-staging"
-version = "0.1.9.202602012332"
+version = "0.1.9.202602020926"
 description = "Bootstrap runner for Flywheel provisioned GPU instances"
 readme = "README.md"
 requires-python = ">=3.11"

{flywheel_bootstrap_staging-0.1.9.202602012332 → flywheel_bootstrap_staging-0.1.9.202602020926}/tests/test_git_ops.py

@@ -234,12 +234,17 @@ class TestSetupBranch:
         mock_run_git.side_effect = [
             MagicMock(returncode=0),  # fetch
             MagicMock(returncode=0, stdout=branch),  # ls-remote (branch exists)
+            MagicMock(returncode=0),  # fetch branch
             MagicMock(returncode=0),  # checkout -B
         ]
 
         result = setup_branch(git_config)
         assert result is True
 
+        call_args_list = [call.args[0] for call in mock_run_git.call_args_list]
+        assert ["fetch", "origin", branch] in call_args_list
+        assert ["checkout", "-B", branch, "FETCH_HEAD"] in call_args_list
+
 
 class TestInitializeRepo:
     """Tests for initialize_repo (integration)."""