From 3242926b4ef27a0a5e761ca1f8d4be7d34c1e7ff Mon Sep 17 00:00:00 2001 From: "Mitch Harding (the weird one)" Date: Tue, 1 Oct 2024 12:50:35 -0400 Subject: [PATCH] Add type hints where missing in IMS client --- src/bos/operators/utils/clients/ims.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/bos/operators/utils/clients/ims.py b/src/bos/operators/utils/clients/ims.py index dbd75c21..1c940fbe 100644 --- a/src/bos/operators/utils/clients/ims.py +++ b/src/bos/operators/utils/clients/ims.py @@ -91,7 +91,7 @@ def get_image(image_id: str, session: RequestsSession|None=None) -> dict: raise -def patch_image(image_id, data, session=None): +def patch_image(image_id: str, data: dict, session: RequestsSession|None=None) -> None: if not data: LOGGER.warning("patch_image called without data; returning without action.") return @@ -111,7 +111,8 @@ def patch_image(image_id, data, session=None): raise -def tag_image(image_id: str, operation: str, key: str, value: str = None, session=None) -> None: +def tag_image(image_id: str, operation: str, key: str, value: str=None, + session: RequestsSession|None=None) -> None: if operation not in IMS_TAG_OPERATIONS: msg = f"{operation} not valid. Expecting one of {IMS_TAG_OPERATIONS}" LOGGER.error(msg)