update: missing refinement

bigcode-project · May 3, 2024 · 8653e0a · 8653e0a
1 parent cab26d2
commit 8653e0a
Show file tree

Hide file tree

Showing 5 changed files with 723 additions and 747 deletions.
diff --git a/data/clean/f_1736_hanhu.py b/data/clean/f_1736_hanhu.py
@@ -29,20 +29,12 @@ def f_1737():
     >>> type(fig).__name__
     'Figure'
     """
-    # Set the font to Arial
-    plt.rcParams['font.family'] = 'Arial'
-
-    # Load the diabetes dataset
-    diabetes = load_diabetes()
-    df = pd.DataFrame(data=diabetes.data, columns=diabetes.feature_names)
-
-    # Create a pairplot
-    pairplot = sns.pairplot(df)
-
-    # Show the plot
-    plt.show()
-
-    return pairplot.fig, df
+    font = {'family': 'Arial'}
+    plt.rc('font', **font)  # Set the global font to Arial.
+    DIABETES = load_diabetes()
+    diabetes_df = pd.DataFrame(data=DIABETES.data, columns=DIABETES.feature_names)
+    pair_plot = sns.pairplot(diabetes_df)
+    return pair_plot.fig, diabetes_df
 
 import unittest
 import matplotlib.pyplot as plt