Fixing FacetGrid figure

tabs/invalid_markets.py

@@ -2,6 +2,7 @@ import pandas as pd
 import gradio as gr
 import matplotlib.pyplot as plt
 import seaborn as sns
+from seaborn import FacetGrid
 
 
 def plot_daily_dist_invalid_trades(invalid_trades: pd.DataFrame):
@@ -37,11 +38,11 @@ def plot_ratio_invalid_trades_per_market(invalid_trades: pd.DataFrame):
 
     # add the IDs as a new column to the original dataframe
     invalid_trades["title_id"] = codes
-    plot = sns.displot(invalid_trades, x="title_id")
+    plot: FacetGrid = sns.displot(invalid_trades, x="title_id")
     plt.xlabel("market id")
     plt.ylabel("Total number of invalid trades by market")
     plt.title("Distribution of invalid trades per market")
-    return gr.Plot(value=plot.get_figure())
+    return gr.Plot(value=plot.figure())
 
 
 def plot_top_invalid_markets(invalid_trades: pd.DataFrame):