fixed pip reference in aws_lambda.py

chompy · chompy · commit d0068a6a07c1 · 2018-04-26T19:28:39.000-04:00
diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
@@ -14,6 +14,11 @@
 import pip
 import yaml
 
+try: # for pip >= 10
+    from pip._internal.operations.freeze import freeze
+except ImportError: # for pip <= 9.0.3
+    from pip.operations.freeze import freeze
+
 from .helpers import mkdir, read, archive, timestamp
 
 
@@ -307,7 +312,7 @@ def pip_install_to_target(path, requirements=False, local_package=None):
     packages = []
     if not requirements:
         print('Gathering pip packages')
-        packages.extend(pip.operations.freeze.freeze())
+        packages.extend(freeze())
     else:
         if os.path.exists("requirements.txt"):
             print('Gathering requirement packages')
