Fix a few more missing __module__s

Author: henryiii

boost_histogram/_internal/axis.py

@@ -119,6 +119,9 @@ def widths(self):
         return self._ax.widths
 
 
+Axis.__module__ = "boost_histogram.axis"
+
+
 class Regular(Axis):
     __slots__ = ()
     _CLASSES = {
@@ -195,6 +198,9 @@ def __init__(self, bins, start, stop, **kwargs):
             raise KeyError("Unsupported collection of options")
 
 
+Regular.__module__ = "boost_histogram.axis"
+
+
 class Variable(Axis):
     __slots__ = ()
     _CLASSES = {
@@ -245,6 +251,9 @@ def __init__(self, edges, **kwargs):
             raise KeyError("Unsupported collection of options")
 
 
+Variable.__module__ = "boost_histogram.axis"
+
+
 class Integer(Axis):
     __slots__ = ()
     _CLASSES = {
@@ -298,6 +307,9 @@ def __init__(self, start, stop, **kwargs):
             raise KeyError("Unsupported collection of options")
 
 
+Integer.__module__ = "boost_histogram.axis"
+
+
 class Category(Axis):
     __slots__ = ()
     _CLASSES = {
@@ -346,6 +358,9 @@ def __init__(self, categories, **kwargs):
             raise KeyError("Unsupported collection of options")
 
 
+Category.__module__ = "boost_histogram.axis"
+
+
 def _to_axis(ax):
     for base in _walk_subclasses(Axis):
         if ax.__class__ in base._CLASSES:

boost_histogram/_internal/hist.py

@@ -231,7 +231,7 @@ def _storage_type(self):
         return _to_storage(self._hist._storage_type)
 
 
-class BoostHistogram(BaseHistogram):
+class histogram(BaseHistogram):
     axis = BaseHistogram._axis
 
     def rank(self):

boost_histogram/cpp/__init__.py

@@ -4,7 +4,7 @@
 
 __all__ = ("histogram", "axis", "storage", "accumulators", "algorithm", "__version__")
 
-from .._internal.hist import BoostHistogram as histogram
+from .._internal.hist import histogram
 
 histogram.__module__ = "boost_histogram.cpp"