avoid error if only one class present in result

WorldCereal / worldcereal-classification

This repository contains the classification module of the WorldCereal system.

https://esa-worldcereal.org/

MIT License

37 stars 4 forks source link

avoid error if only one class present in result #208

Closed jdegerickx closed 1 month ago

jdegerickx commented 1 month ago

upon testing visualization, I ran into an error when there is only one class in the raster. This fixes the issue. Also tested with multi-class crop type output and seems to work fine.