fix KeyError in os.environ check

[arvados.git] / sdk / python / arvados.py

diff --git a/sdk/python/arvados.py b/sdk/python/arvados.py
index 36a1ea2f0931fe36d55bd0131d344d1ac4d2867c..9ba435e10379eb6596be1cbde51d5727c3712e76 100644 (file)
--- a/sdk/python/arvados.py
+++ b/sdk/python/arvados.py
@@ -384,7 +384,7 @@ class CollectionWriter:
 class Keep:
     @staticmethod
     def put(data):
-        if os.environ['KEEP_LOCAL_STORE']:
+        if 'KEEP_LOCAL_STORE' in os.environ:
             return Keep.local_store_put(data)
         p = subprocess.Popen(["whput", "-"],
                              stdout=subprocess.PIPE,
@@ -397,7 +397,7 @@ class Keep:
         return stdoutdata.rstrip()
     @staticmethod
     def get(locator):
-        if os.environ['KEEP_LOCAL_STORE']:
+        if 'KEEP_LOCAL_STORE' in os.environ:
             return Keep.local_store_get(locator)
         p = subprocess.Popen(["whget", locator, "-"],
                              stdout=subprocess.PIPE,