Made some code a lot cleaner

CozySynthesizer · Jul 19, 2018 · c7ca050 · c7ca050
1 parent 240df72
commit c7ca050
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/cozy/cost_model.py b/cozy/cost_model.py
@@ -341,13 +341,7 @@ def _maintenance_cost(e : Exp, solver : ModelCachingSolver, op : Op, freebies :
 
 def maintenance_cost(e : Exp, solver : ModelCachingSolver, op : Op, freebies : [Exp] = []):
     """This method calulates the result over all expressions that are EStateVar """
-    res = ZERO
-    for x in all_exps(e):
-        if isinstance(x, EStateVar):
-            res = ESum([
-                res,
-                _maintenance_cost(e=x.e, solver=solver, op=op, freebies=freebies)])
-    return res
+    return ESum([_maintenance_cost(e=x.e, solver=solver, op=op, freebies=freebies) for x in all_exps(e) if isinstance(x, EStateVar)])
 
 
 # These require walking over the entire collection.