Update NumPy import

kgryte · kgryte · commit 05f633872770 · 2017-10-11T10:30:53.000-07:00
diff --git a/tools/snippets/benchmark/python/numpy/benchmark.py b/tools/snippets/benchmark/python/numpy/benchmark.py
@@ -55,8 +55,8 @@ def print_results(elapsed):
 
 def benchmark():
     """Run the benchmark and print benchmark results."""
-    setup = "from numpy import TODO; from random import random;"
-    stmt = "y = TODO(random())"
+    setup = "import numpy as np; from random import random;"
+    stmt = "y = np.TODO(random())"
 
     t = timeit.Timer(stmt, setup=setup)
 
