add perfect forecat case to RBLQ

QuantEcon · Oct 21, 2017 · 05cbb3c · 05cbb3c
1 parent 899358f
commit 05cbb3c
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/quantecon/robustlq.py b/quantecon/robustlq.py
@@ -75,10 +75,7 @@ def __init__(self, Q, R, A, B, C, beta, theta):
         # == Remaining parameters == #
         self.beta, self.theta = beta, theta
         # == Check for case of no control (pure forecasting problem) == #
-        if not Q.any() and not B.any():
-            self.pure_forecasting = True
-        else:
-            self.pure_forecasting = False
+        self.pure_forecasting = True if not Q.any() and not B.any() else False
 
     def __repr__(self):
         return self.__str__()
@@ -191,6 +188,7 @@ def robust_rule(self):
         # == Set up LQ version == #
         I = identity(j)
         Z = np.zeros((k, j))
+
         if self.pure_forecasting:
             lq = LQ(-beta*I*theta, R, A, C, beta=beta)