(keras image) Assert that we have 1 target in explanation only

TeamHG-Memex · Aug 5, 2019 · 0a4db99 · 0a4db99
1 parent 87cafea
commit 0a4db99
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/eli5/formatters/image.py b/eli5/formatters/image.py
@@ -104,6 +104,7 @@ def format_as_image(expl, # type: Explanation
         # no heatmaps
         return image
     else:
+        assert len(expl.targets) == 1
         heatmap = expl.targets[0].heatmap
         _validate_heatmap(heatmap)