FIX/#199

mabel-dev · Jun 18, 2022 · 0f5e7f5 · 0f5e7f5
1 parent ca8d138
commit 0f5e7f5
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/opteryx/engine/planner/operations/show_columns.py b/opteryx/engine/planner/operations/show_columns.py
@@ -136,7 +136,7 @@ def _extended_collector(pages):
     """
     Collect summary statistics about each column
     """
-    from opteryx.sketches.counting_tree import CountingTree
+    from opteryx.sketches import CountingTree
     from opteryx.third_party import distogram
     from opteryx.third_party import hyperloglog
 

diff --git a/opteryx/sketches/__init__.py b/opteryx/sketches/__init__.py
@@ -1,3 +1,5 @@
 import pyximport
 
 pyximport.install()
+
+from .counting_tree import CountingTree
diff --git a/opteryx/version.py b/opteryx/version.py
@@ -16,4 +16,4 @@
 2) we can import it in setup.py for the same reason
 """
 
-__version__ = "0.0.3-beta.9"
+__version__ = "0.0.3-beta.10"
diff --git a/tests/sketches/test_counting_tree.py b/tests/sketches/test_counting_tree.py
@@ -3,7 +3,7 @@
 
 sys.path.insert(1, os.path.join(sys.path[0], "../.."))
 
-from opteryx.sketches.counting_tree import CountingTree
+from opteryx.sketches import CountingTree
 
 
 def test_counter():