Skip to content

Commit

Permalink
update base geti class balance call
Browse files Browse the repository at this point in the history
Signed-off-by: Igor Davidyuk <[email protected]>
  • Loading branch information
igor-davidyuk committed Jul 30, 2024
1 parent 3b2d19f commit 6b00005
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions geti_sdk/geti.py
Original file line number Diff line number Diff line change
Expand Up @@ -207,11 +207,12 @@ def projects(self) -> List[Project]:
@property
def credit_balance(self) -> Optional[int]:
"""
Get the current credit balance in the workspace.
Get the current available credit balance in the workspace.
:return: The available credit balance in the workspace.
"""
return self.credit_system_client.get_balance()
balance = self.credit_system_client.get_balance()
return balance.available if balance is not None else None

def get_project(
self, project_name: str, project_id: Optional[str] = None
Expand Down

0 comments on commit 6b00005

Please sign in to comment.