Add aggregation as top-level import

dcherian · dcherian · commit 96c6b49418fe · 2022-03-31T10:19:24.000-07:00
diff --git a/docs/source/api.rst b/docs/source/api.rst
@@ -39,6 +39,6 @@ Aggregation Objects
 .. autosummary::
     :toctree: generated/
 
-    aggregations.Aggregation
+    Aggregation
     aggregations.sum_
     aggregations.nansum
diff --git a/docs/source/user-stories/custom-aggregations.ipynb b/docs/source/user-stories/custom-aggregations.ipynb
@@ -665,7 +665,7 @@
     }
    ],
    "source": [
-    "from flox.aggregations import Aggregation\n",
+    "from flox import Aggregation\n",
     "\n",
     "agg_median = Aggregation(\n",
     "    name=\"median\",\n",
diff --git a/flox/__init__.py b/flox/__init__.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 # flake8: noqa
 """Top-level module for flox ."""
+from .aggregations import Aggregation  # noqa
 from .core import groupby_reduce, rechunk_for_blockwise, rechunk_for_cohorts  # noqa
 
 try:

Original file line number	Diff line number	Diff line change
`@@ -665,7 +665,7 @@`
`665`	`665`	`}`
`666`	`666`	`],`
`667`	`667`	`"source": [`
`668`		`- "from flox.aggregations import Aggregation\n",`
	`668`	`+ "from flox import Aggregation\n",`
`669`	`669`	`"\n",`
`670`	`670`	`"agg_median = Aggregation(\n",`
`671`	`671`	`" name=\"median\",\n",`