s/"main"/MAIN_BRANCH

kevinjqliu · kevinjqliu · commit ad1ec672bac7 · 2024-01-27T10:46:37.000-08:00
diff --git a/pyiceberg/table/__init__.py b/pyiceberg/table/__init__.py
@@ -269,7 +269,7 @@ def set_ref_snapshot(
             )
         )
 
-        self._append_requirements(AssertRefSnapshotId(snapshot_id=parent_snapshot_id, ref="main"))
+        self._append_requirements(AssertRefSnapshotId(snapshot_id=parent_snapshot_id, ref=MAIN_BRANCH))
         return self
 
     def update_schema(self) -> UpdateSchema:
@@ -2445,7 +2445,7 @@ def commit(self) -> Snapshot:
         with self._table.transaction() as tx:
             tx.add_snapshot(snapshot=snapshot)
             tx.set_ref_snapshot(
-                snapshot_id=self._snapshot_id, parent_snapshot_id=self._parent_snapshot_id, ref_name="main", type="branch"
+                snapshot_id=self._snapshot_id, parent_snapshot_id=self._parent_snapshot_id, ref_name=MAIN_BRANCH, type="branch"
             )
 
         return snapshot

Original file line number	Diff line number	Diff line change
`@@ -269,7 +269,7 @@ def set_ref_snapshot(`
`269`	`269`	`)`
`270`	`270`	`)`
`271`	`271`
`272`		`- self._append_requirements(AssertRefSnapshotId(snapshot_id=parent_snapshot_id, ref="main"))`
	`272`	`+ self._append_requirements(AssertRefSnapshotId(snapshot_id=parent_snapshot_id, ref=MAIN_BRANCH))`
`273`	`273`	`return self`
`274`	`274`
`275`	`275`	`def update_schema(self) -> UpdateSchema:`
`@@ -2445,7 +2445,7 @@ def commit(self) -> Snapshot:`
`2445`	`2445`	`with self._table.transaction() as tx:`
`2446`	`2446`	`tx.add_snapshot(snapshot=snapshot)`
`2447`	`2447`	`tx.set_ref_snapshot(`
`2448`		`- snapshot_id=self._snapshot_id, parent_snapshot_id=self._parent_snapshot_id, ref_name="main", type="branch"`
	`2448`	`+ snapshot_id=self._snapshot_id, parent_snapshot_id=self._parent_snapshot_id, ref_name=MAIN_BRANCH, type="branch"`
`2449`	`2449`	`)`
`2450`	`2450`
`2451`	`2451`	`return snapshot`