fixed logic for model collection indexing with multi-target

erikerlandson · Jun 17, 2012 · 0166a00 · 0166a00
1 parent 9351eec
commit 0166a00
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bin/score_models b/bin/score_models
@@ -73,7 +73,8 @@ if args.scores is not None:
 sys.stdout.write("indexing data...\n")
 data.ensure_index([(args.cvattr, pymongo.ASCENDING)])
 data.ensure_index([(args.user, pymongo.ASCENDING)])
-models.ensure_index([("_id.y", pymongo.ASCENDING), ("rr", pymongo.DESCENDING)])
+for t in args.target:
+    t['coll'].ensure_index([("_id.y", pymongo.ASCENDING), ("rr", pymongo.DESCENDING)])
 
 # some scoring stats
 nall = 0