escape errors in swarmplot

hakuhodo-technologies · Jul 23, 2023 · ea029c2 · ea029c2
1 parent e823f6f
commit ea029c2
Showing 1 changed file with 73 additions and 52 deletions.
diff --git a/scope_rl/ope/ope.py b/scope_rl/ope/ope.py
@@ -2059,14 +2059,17 @@ def visualize_policy_value_with_multiple_estimates(
                         )
 
                     else:
-                        sns.swarmplot(
-                            data=df,
-                            x="estimator",
-                            y="policy_value",
-                            hue="behavior_policy",
-                            palette=palette,
-                            ax=ax,
-                        )
+                        try:
+                            sns.swarmplot(
+                                data=df,
+                                x="estimator",
+                                y="policy_value",
+                                hue="behavior_policy",
+                                palette=palette,
+                                ax=ax,
+                            )
+                        except:
+                            warn("Encountered NaN values during plot.")
 
                     on_policy = policy_value_dict[behavior_policy][eval_policy][
                         "on_policy"
@@ -2272,13 +2275,16 @@ def visualize_policy_value_with_multiple_estimates(
                             ax=ax,
                         )
                     else:
-                        sns.swarmplot(
-                            data=df,
-                            x="estimator",
-                            y="policy_value",
-                            palette=palette,
-                            ax=ax,
-                        )
+                        try:
+                            sns.swarmplot(
+                                data=df,
+                                x="estimator",
+                                y="policy_value",
+                                palette=palette,
+                                ax=ax,
+                            )
+                        except:
+                            warn("Encountered NaN values during plot.")
 
                     on_policy = policy_value_dict[eval_policy]["on_policy"]
                     if on_policy is not None:
@@ -2357,13 +2363,16 @@ def visualize_policy_value_with_multiple_estimates(
                         )
 
                     else:
-                        sns.swarmplot(
-                            data=df,
-                            x="eval_policy",
-                            y="policy_value",
-                            palette=palette,
-                            ax=ax,
-                        )
+                        try:
+                            sns.swarmplot(
+                                data=df,
+                                x="eval_policy",
+                                y="policy_value",
+                                palette=palette,
+                                ax=ax,
+                            )
+                        except:
+                            warn("Encountered NaN values during plot.")
 
                     if visualize_on_policy:
                         ax.scatter(
@@ -5530,14 +5539,17 @@ def _visualize_off_policy_estimates_with_multiple_estimates(
                         )
 
                     else:
-                        sns.swarmplot(
-                            data=df,
-                            x="estimator",
-                            y="policy_value",
-                            hue="behavior_policy",
-                            palette=palette,
-                            ax=ax,
-                        )
+                        try:
+                            sns.swarmplot(
+                                data=df,
+                                x="estimator",
+                                y="policy_value",
+                                hue="behavior_policy",
+                                palette=palette,
+                                ax=ax,
+                            )
+                        except:
+                            warn("Encountered NaN values during plot.")
 
                     on_policy = estimation_dict[behavior_policy][eval_policy][
                         "on_policy"
@@ -5645,14 +5657,17 @@ def _visualize_off_policy_estimates_with_multiple_estimates(
                         )
 
                     else:
-                        sns.swarmplot(
-                            data=df,
-                            x="eval_policy",
-                            y="policy_value",
-                            hue="behavior_policy",
-                            palette=palette,
-                            ax=ax,
-                        )
+                        try:
+                            sns.swarmplot(
+                                data=df,
+                                x="eval_policy",
+                                y="policy_value",
+                                hue="behavior_policy",
+                                palette=palette,
+                                ax=ax,
+                            )
+                        except:
+                            warn("Encountered NaN values during plot.")
 
                     if visualize_on_policy:
                         ax.scatter(
@@ -5740,13 +5755,16 @@ def _visualize_off_policy_estimates_with_multiple_estimates(
                             ax=ax,
                         )
                     else:
-                        sns.swarmplot(
-                            data=df,
-                            x="estimator",
-                            y="policy_value",
-                            palette=palette,
-                            ax=ax,
-                        )
+                        try:
+                            sns.swarmplot(
+                                data=df,
+                                x="estimator",
+                                y="policy_value",
+                                palette=palette,
+                                ax=ax,
+                            )
+                        except:
+                            warn("Encountered NaN values during plot.")
 
                     on_policy = estimation_dict[eval_policy]["on_policy"]
                     if on_policy is not None:
@@ -5838,13 +5856,16 @@ def _visualize_off_policy_estimates_with_multiple_estimates(
                         )
 
                     else:
-                        sns.swarmplot(
-                            data=df,
-                            x="eval_policy",
-                            y="policy_value",
-                            palette=palette,
-                            ax=ax,
-                        )
+                        try:
+                            sns.swarmplot(
+                                data=df,
+                                x="eval_policy",
+                                y="policy_value",
+                                palette=palette,
+                                ax=ax,
+                            )
+                        except:
+                            warn("Encountered NaN values during plot.")
 
                     if visualize_on_policy:
                         ax.scatter(