Set column types explicitly

jnothman · Dec 27, 2023 · 16e3d68 · 16e3d68
1 parent 5e970e8
commit 16e3d68
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/upsetplot/plotting.py b/upsetplot/plotting.py
@@ -656,7 +656,9 @@ def plot_matrix(self, ax):
                          "linewidth": "linewidths",
                          "linestyle": "linestyles",
                          "hatch": "hatch"}
-        styles = pd.DataFrame(styles).reindex(columns=style_columns.keys())
+        styles = (pd.DataFrame(styles)
+                  .reindex(columns=style_columns.keys())
+                  .astype(['o', 'o', int, 'o', 'o']))
         styles["linewidth"].fillna(1, inplace=True)
         styles["facecolor"].fillna(self._facecolor, inplace=True)
         styles["edgecolor"].fillna(styles["facecolor"], inplace=True)