Fix benchmark extension module import (#428)

msgpack · Jun 22, 2020 · 3508ca5 · 3508ca5
1 parent c1b1a23
commit 3508ca5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/benchmark/benchmark.py b/benchmark/benchmark.py
@@ -1,7 +1,7 @@
 from msgpack import fallback
 
 try:
-    from msgpack import _unpacker, _packer
+    from msgpack import _cmsgpack
 
     has_ext = True
 except ImportError:
@@ -17,14 +17,14 @@ def profile(name, func):
 
 def simple(name, data):
     if has_ext:
-        packer = _packer.Packer()
+        packer = _cmsgpack.Packer()
         profile("packing %s (ext)" % name, lambda: packer.pack(data))
     packer = fallback.Packer()
     profile("packing %s (fallback)" % name, lambda: packer.pack(data))
 
     data = packer.pack(data)
     if has_ext:
-        profile("unpacking %s (ext)" % name, lambda: _unpacker.unpackb(data))
+        profile("unpacking %s (ext)" % name, lambda: _cmsgpack.unpackb(data))
     profile("unpacking %s (fallback)" % name, lambda: fallback.unpackb(data))