From 38295c88b5b47c2c7cedd98ef2fc2f505f11551a Mon Sep 17 00:00:00 2001
From: michaelhtm <98621731+michaelhtm@users.noreply.github.com>
Date: Thu, 31 Jul 2025 11:47:14 -0700
Subject: [PATCH] chore: add partition to acktest identity functions

---
 src/acktest/aws/identity.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/acktest/aws/identity.py b/src/acktest/aws/identity.py
index f1f28e3e..b5b9cd2d 100644
--- a/src/acktest/aws/identity.py
+++ b/src/acktest/aws/identity.py
@@ -22,3 +22,6 @@ def get_account_id() -> int:
 
 def get_region(default: str = "us-west-2") -> str:
     return boto3.session.Session().region_name or default
+
+def get_partition() -> str:
+    return boto3.client('sts').get_caller_identity().get('Arn').split(':')[1]
\ No newline at end of file