Fix code completion

Author: FBumann

docs/examples/advanced.ipynb

@@ -17,8 +17,9 @@
    "source": [
     "import numpy as np\n",
     "import pandas as pd\n",
+    "import xarray as xr\n",
     "\n",
-    "from xarray_plotly import DataArray  # IDE completion for .plotly"
+    "import xarray_plotly"
    ]
   },
   {
@@ -36,7 +37,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "da = DataArray(\n",
+    "da = xr.DataArray(\n",
     "    np.random.randn(30, 3).cumsum(axis=0) + 15,\n",
     "    dims=[\"time\", \"station\"],\n",
     "    coords={\n",
@@ -376,7 +377,7 @@
     "fig = monthly_mean.plotly.line(\n",
     "    title=\"Monthly Climatology\",\n",
     ")\n",
-    "fig.update_xaxes(tickmode=\"array\", tickvals=list(range(1, 13)),\n",
+    "fig.update_xaxes(tickmode=\"array\", tickvals=list(range(1, 13)), \n",
     "                 ticktext=[\"Jan\", \"Feb\", \"Mar\", \"Apr\", \"May\", \"Jun\",\n",
     "                          \"Jul\", \"Aug\", \"Sep\", \"Oct\", \"Nov\", \"Dec\"])\n",
     "fig"

docs/examples/plot-types.ipynb

@@ -17,8 +17,9 @@
    "source": [
     "import numpy as np\n",
     "import pandas as pd\n",
+    "import xarray as xr\n",
     "\n",
-    "from xarray_plotly import DataArray  # IDE completion for .plotly"
+    "import xarray_plotly"
    ]
   },
   {
@@ -39,7 +40,7 @@
     "np.random.seed(42)\n",
     "\n",
     "# Time series data\n",
-    "da_ts = DataArray(\n",
+    "da_ts = xr.DataArray(\n",
     "    np.random.randn(30, 3).cumsum(axis=0),\n",
     "    dims=[\"time\", \"category\"],\n",
     "    coords={\n",
@@ -50,7 +51,7 @@
     ")\n",
     "\n",
     "# 2D grid data\n",
-    "da_2d = DataArray(\n",
+    "da_2d = xr.DataArray(\n",
     "    np.random.rand(20, 30),\n",
     "    dims=[\"lat\", \"lon\"],\n",
     "    coords={\n",
@@ -61,7 +62,7 @@
     ")\n",
     "\n",
     "# Categorical data\n",
-    "da_cat = DataArray(\n",
+    "da_cat = xr.DataArray(\n",
     "    np.random.rand(4, 3) * 100,\n",
     "    dims=[\"product\", \"region\"],\n",
     "    coords={\n",
@@ -160,7 +161,7 @@
    "outputs": [],
    "source": [
     "# Use absolute values for stacking to make sense\n",
-    "da_positive = DataArray(\n",
+    "da_positive = xr.DataArray(\n",
     "    np.abs(np.random.randn(30, 3)) * 10,\n",
     "    dims=[\"time\", \"source\"],\n",
     "    coords={\n",
@@ -228,7 +229,7 @@
    "outputs": [],
    "source": [
     "# Create data with more samples\n",
-    "da_dist = DataArray(\n",
+    "da_dist = xr.DataArray(\n",
     "    np.random.randn(100, 4) + np.array([0, 1, 2, 3]),\n",
     "    dims=[\"sample\", \"group\"],\n",
     "    coords={\"group\": [\"Control\", \"Treatment A\", \"Treatment B\", \"Treatment C\"]},\n",
@@ -294,7 +295,7 @@
    "outputs": [],
    "source": [
     "# 3D data for faceting\n",
-    "da_3d = DataArray(\n",
+    "da_3d = xr.DataArray(\n",
     "    np.random.randn(30, 3, 2).cumsum(axis=0),\n",
     "    dims=[\"time\", \"city\", \"scenario\"],\n",
     "    coords={\n",
@@ -329,11 +330,11 @@
    "outputs": [],
    "source": [
     "# Create monthly data\n",
-    "da_monthly = DataArray(\n",
+    "da_monthly = xr.DataArray(\n",
     "    np.random.rand(12, 4) * 100,\n",
     "    dims=[\"month\", \"product\"],\n",
     "    coords={\n",
-    "        \"month\": [\"Jan\", \"Feb\", \"Mar\", \"Apr\", \"May\", \"Jun\",\n",
+    "        \"month\": [\"Jan\", \"Feb\", \"Mar\", \"Apr\", \"May\", \"Jun\", \n",
     "                  \"Jul\", \"Aug\", \"Sep\", \"Oct\", \"Nov\", \"Dec\"],\n",
     "        \"product\": [\"A\", \"B\", \"C\", \"D\"],\n",
     "    },\n",

docs/getting-started.ipynb

@@ -32,7 +32,7 @@
    "source": [
     "## Basic Usage\n",
     "\n",
-    "Import `DataArray` from xarray_plotly for IDE code completion:"
+    "Import the package to register the `plotly` accessor:"
    ]
   },
   {
@@ -45,7 +45,7 @@
     "import pandas as pd\n",
     "import xarray as xr\n",
     "\n",
-    "import xarray_plotly"
+    "import xarray_plotly  # This registers the accessor"
    ]
   },
   {
@@ -78,7 +78,7 @@
     "    attrs={\"long_name\": \"Temperature Anomaly\", \"units\": \"°C\"},\n",
     ")\n",
     "\n",
-    "da.plotly."
+    "da"
    ]
   },
   {

xarray_plotly/__init__.pyi

@@ -10,6 +10,6 @@ from xarray_plotly.common import auto as auto
 __version__: str
 
 # Augment xarray.DataArray with plotly accessor for IDE autocompletion
-class DataArray(_DataArray):
+class DataArray(_DataArray):  # type: ignore[no-untyped-call]
     @property
     def plotly(self) -> DataArrayPlotlyAccessor: ...