switch cache location

amazon-braket · Oct 12, 2023 · 81b7816 · 81b7816
1 parent 88f832d
commit 81b7816
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/braket/jobs/image_uris.py b/src/braket/jobs/image_uris.py
@@ -26,11 +26,11 @@ class Framework(str, Enum):
     PL_PYTORCH = "PL_PYTORCH"
 
 
-@cache
 def built_in_images(region):
     return {retrieve_image(framework, region) for framework in Framework}
 
 
+@cache
 def retrieve_image(framework: Framework, region: str) -> str:
     """Retrieves the ECR URI for the Docker image matching the specified arguments.