Closes #1200

apache · Oct 31, 2016 · ef01140 · ef01140
2 parents 00b17e5 + 32f3214
commit ef01140
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/sdks/python/apache_beam/utils/dependency.py b/sdks/python/apache_beam/utils/dependency.py
@@ -476,7 +476,13 @@ def _download_pypi_sdk_package(temp_dir):
   """Downloads SDK package from PyPI and returns path to local path."""
   # TODO(silviuc): Handle apache-beam versions when we have official releases.
   import pkg_resources as pkg
-  version = pkg.get_distribution(GOOGLE_PACKAGE_NAME).version
+  try:
+    version = pkg.get_distribution(GOOGLE_PACKAGE_NAME).version
+  except pkg.DistributionNotFound:
+    raise RuntimeError('Please set --sdk_location command-line option '
+                       'or install a valid {} distribution.'
+                       .format(GOOGLE_PACKAGE_NAME))
+
   # Get a source distribution for the SDK package from PyPI.
   cmd_args = [
       _get_python_executable(), '-m', 'pip', 'install', '--download', temp_dir,