def get_masks()

in src/data_utils.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

def get_masks(base_image):
    """Returns image masks corresponding to mangrove and non-mangrove pixels

    Args:
        base_image: earth engine image to create masks from

    Returns:
        objects: ee.Image, ee.Image
    """
    img_mangrove = get_mangrove_data()
    mangrove_mask = base_image.updateMask(img_mangrove.eq(1))
    non_mangrove_mask = base_image.updateMask(mangrove_mask.unmask().Not())

    return mangrove_mask, non_mangrove_mask