Fixes bug from #72, exception included.

nasa-nccs-hpda · Apr 4, 2024 · b0e6467 · b0e6467
1 parent 20917d3
commit b0e6467
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.rst b/README.rst
@@ -219,7 +219,7 @@ an example of manually testing the package components.
 
 .. code:: python
 
-singularity exec --env PYTHONPATH="$NOBACKUP/development/tensorflow-caney:$NOBACKUP/development/vhr-cloudmask" --nv -B $NOBACKUP,/explore/nobackup/people,/explore/nobackup/projects /explore/nobackup/projects/ilab/containers/vhr-cloudmask.sif pytest $NOBACKUP/development/vhr-cloudmask/tests
+        singularity exec --env PYTHONPATH="$NOBACKUP/development/tensorflow-caney:$NOBACKUP/development/vhr-cloudmask" --nv -B $NOBACKUP,/explore/nobackup/people,/explore/nobackup/projects /explore/nobackup/projects/ilab/containers/vhr-cloudmask.sif pytest $NOBACKUP/development/vhr-cloudmask/tests
 
 Authors
 ====================

diff --git a/vhr_cloudmask/model/pipelines/cloudmask_cnn_pipeline.py b/vhr_cloudmask/model/pipelines/cloudmask_cnn_pipeline.py
@@ -495,13 +495,13 @@ def calc_cloud_percentage(self, prediction: np.ndarray) -> float:
         # get cloudy pixels
         try:
             non_cloud_pixels = unique_dict[0]
-        except IndexError:
+        except (IndexError, KeyError):
             non_cloud_pixels = 0
 
         # get cloudy pixels
         try:
             cloud_pixels = unique_dict[1]
-        except IndexError:
+        except (IndexError, KeyError):
             cloud_pixels = 0
 
         # percent cloud cover