Minor code cleanup for strat code.

bloom-lang · Jan 16, 2014 · f835046 · f835046
1 parent 92771a2
commit f835046
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/bud/bud_meta.rb b/lib/bud/bud_meta.rb
@@ -16,10 +16,12 @@ def meta_rewrite
 
     stratified_rules = []
     if @bud_instance.toplevel == @bud_instance
-      stratum_map = stratify_preds
+      stratum_map, nodes = stratify_preds
       top_stratum = stratum_map.values.max
       top_stratum ||= -1
 
+      analyze_dependencies(nodes)
+
       # stratum_map = {fully qualified pred => stratum}. Copy stratum_map data
       # into t_stratum format.
       raise unless @bud_instance.t_stratum.to_a.empty?
@@ -227,8 +229,7 @@ def stratify_preds
       n.stratum = remap[n.stratum]
       stratum_map[n.name] = n.stratum
     end
-    analyze_dependencies(nodes)
-    return stratum_map
+    return stratum_map, nodes
   end
 
   def calc_stratum(node, neg, temporal, path)