fix for len(dim)==1

observingClouds · Jun 24, 2023 · 63f1809 · 63f1809
1 parent 6597166
commit 63f1809
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xbitinfo/graphics.py b/xbitinfo/graphics.py
@@ -233,6 +233,7 @@ def plot_bitinformation(bitinfo, cmap="turku", crop=None):
     """
     import matplotlib.pyplot as plt
 
+    bitinfo = bitinfo.squeeze()
     assert (
         "dim" not in bitinfo.dims
     ), "Found dependence of bitinformation on dimension. Please reduce data first by e.g. `bitinfo.max(dim='dim')`"