Check for ImportError in benchmarks/plasma.py

Change-Id: I6e6779b656e2e18350e6d2de82edd6239a7b915c

python/benchmarks/plasma.py

@@ -18,8 +18,12 @@
 import numpy as np
 import timeit
 
-import pyarrow.plasma as plasma
-
+try:
+    import pyarrow.plasma as plasma
+except ImportError:
+    # TODO(wesm): These are not asv benchmarks, so we can just fail
+    # silently here
+    pass
 
 class SimplePlasmaThroughput(object):
     """Benchmark plasma store throughput with a single client."""

python/pyarrow/plasma.py

@@ -24,15 +24,8 @@ import sys
 import tempfile
 import time
 
-
-try:
-    from pyarrow._plasma import (ObjectID, ObjectNotAvailable, # noqa
-                                 PlasmaBuffer, PlasmaClient, connect)
-except ImportError:
-    # TODO(wesm): These are not asv benchmarks, so we can just fail
-    # silently here
-    pass
-
+from pyarrow._plasma import (ObjectID, ObjectNotAvailable, # noqa
+                             PlasmaBuffer, PlasmaClient, connect)
 
 # The Plasma TensorFlow Operator needs to be compiled on the end user's
 # machine since the TensorFlow ABI is not stable between versions.