Fix bug on seaborn plot

RussellDash332 · Jan 31, 2024 · 2adabe8 · 2adabe8
1 parent 056356f
commit 2adabe8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/autokattis/api/__init__.py b/autokattis/api/__init__.py
@@ -184,10 +184,12 @@ def plot_problems(self, filepath=None, show_solved=True, show_partial=True, show
         palette = {'Easy': '#39a137', 'Medium': '#ffbe00', 'Hard': '#ff411a', 'N/A': 'gray'}
         for c in [*palette]:
             if c not in categories: del palette[c]
-        hist = sns.histplot(data=df, x='difficulty', hue='category', multiple='stack', binwidth=0.1, hue_order=hue_order, palette=palette)
+        diff_lo = math.floor(min([d for d in df.difficulty if d], default=0))
+        diff_hi = math.ceil(max([d for d in df.difficulty if d], default=0))
+        hist = sns.histplot(data=df, x='difficulty', hue='category', multiple='stack', bins=[i/10 for i in range(10*diff_lo, 10*diff_hi+1)], hue_order=hue_order, palette=palette)
         hist.set(title=f'Solved Kattis Problems by {self.user} ({df.shape[0]})', xlabel='Difficulty')
         plt.legend(title='Category', loc='upper right', labels=hue_order[::-1])
-        plt.xticks([*range(math.floor(min([d for d in df.difficulty if d], default=0)), math.ceil(max([d for d in df.difficulty if d], default=0))+1)])
+        plt.xticks([*range(diff_lo, diff_hi+1)])
         if filepath != None:
             plt.savefig(filepath)
             print(f'Saved to {filepath}')

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup_args = dict(
     name='autokattis',
-    version='1.6.1',
+    version='1.6.2',
     description='Updated Kattis API wrapper',
     long_description_content_type="text/markdown",
     long_description=README,