regression: release argument wasn't mandatory for this call

pitthsls · Mar 28, 2019 · 852065e · 852065e
1 parent b57ada5
commit 852065e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pycounter/sushi.py b/pycounter/sushi.py
@@ -14,11 +14,12 @@
 import requests
 import six
 
+from pycounter import sushi5
 import pycounter.constants
 import pycounter.exceptions
 from pycounter.helpers import convert_date_run
 import pycounter.report
-from pycounter import sushi5
+
 
 logger = logging.getLogger(__name__)
 NS = pycounter.constants.NS
@@ -131,7 +132,7 @@ def get_report(*args, **kwargs):
 
     :param no_delay: don't delay in retrying Report Queued
     """
-    if kwargs["release"] == 5:
+    if kwargs.get("release") == 5:
         return sushi5.get_report(*args, **kwargs)
 
     no_delay = kwargs.pop("no_delay", False)