From 963b1f30a0ac1a3a3abf49713a9f0d08085f6efe Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Fri, 19 Jan 2024 17:24:30 -0500 Subject: [PATCH] Update branch methods' docstrings to include branch_id --- neon_client/client.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/neon_client/client.py b/neon_client/client.py index d0f6596..164a53b 100644 --- a/neon_client/client.py +++ b/neon_client/client.py @@ -152,20 +152,21 @@ class NeonAPI: def branch_update( self, project_id: str, branch_id: str, **json: dict ) -> t.Dict[str, t.Any]: - """Update a branch. Accepts all keyword arguments for json body.""" + """Update a branch by branch_id. Accepts all keyword arguments for json body.""" return self.request( "PATCH", f"projects/{ project_id }/branches/{ branch_id }", json=json ) def branch_delete(self, project_id: str, branch_id: str) -> t.Dict[str, t.Any]: - """Delete a branch.""" + """Delete a branch by branch_id.""" return self.request("DELETE", f"projects/{ project_id }/branches/{ branch_id }") def branch_set_as_primary( self, project_id: str, branch_id: str ) -> t.Dict[str, t.Any]: - """Set a branch as primary.""" + """Set a branch as primary by branch_id.""" + return self.request( "POST", f"projects/{ project_id }/branches/{ branch_id }/set_as_primary" )