fix: ensure xticks are within plot limits (#125)

Nixtla · Sep 23, 2024 · 73be061 · 73be061
1 parent 6d06cb1
commit 73be061
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 16 deletions.
diff --git a/nbs/imgs/index.png b/nbs/imgs/index.png
diff --git a/nbs/imgs/plotting.png b/nbs/imgs/plotting.png
diff --git a/nbs/plotting.ipynb b/nbs/plotting.ipynb
@@ -443,14 +443,7 @@
     "                ax[row, col].set_ylabel(ylabel)\n",
     "            if row == n_rows - 1:\n",
     "                ax[row, col].set_xlabel(xlabel)\n",
-    "            xticklabels = ax[row, col].get_xticklabels()\n",
-    "            xticks = ax[row, col].get_xticks()\n",
-    "            ax[row, col].set_xticks(\n",
-    "                rotation=30, \n",
-    "                ticks=xticks, \n",
-    "                labels=xticklabels,\n",
-    "                ha=\"right\",\n",
-    "            )\n",
+    "            ax[row, col].tick_params(axis='x', labelrotation=30)\n",
     "        else:\n",
     "            fig.update_annotations(selector={\"text\": str(i)}, text=title)\n",
     "\n",

diff --git a/utilsforecast/plotting.py b/utilsforecast/plotting.py
@@ -389,14 +389,7 @@ def _add_plotly_plot(fig, df, y_col, levels):
                 ax[row, col].set_ylabel(ylabel)
             if row == n_rows - 1:
                 ax[row, col].set_xlabel(xlabel)
-            xticklabels = ax[row, col].get_xticklabels()
-            xticks = ax[row, col].get_xticks()
-            ax[row, col].set_xticks(
-                rotation=30,
-                ticks=xticks,
-                labels=xticklabels,
-                ha="right",
-            )
+            ax[row, col].tick_params(axis="x", labelrotation=30)
         else:
             fig.update_annotations(selector={"text": str(i)}, text=title)