From a077a7f39f1077ace9cb97401cee0fede889a527 Mon Sep 17 00:00:00 2001
From: cxchu <cxchu@lap-15-71>
Date: Mon, 27 Apr 2020 18:28:54 +0200
Subject: [PATCH] ignore occurrence loading in big universes

---
 typeConsolidation/TypeConsolidation.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/typeConsolidation/TypeConsolidation.py b/typeConsolidation/TypeConsolidation.py
index 0d83805..3f52ff8 100644
--- a/typeConsolidation/TypeConsolidation.py
+++ b/typeConsolidation/TypeConsolidation.py
@@ -222,11 +222,13 @@ def optimization(mention, candidateTypes, type2score, disjointContr, hierarchica
 print('\t\tloading occurrence.....')   
 sys.stdout.flush() 
 pearsonValues = {}
-
+big_universes = {'camphalfbloodroleplay', 'fallout', 'enmarveldatabase', 
+                'forgottenrealms', 'harrypotter', 'leagueoflegends', 'lego', 
+                'psychology', 'starwars', 'villains', 'wowwiki', 'xenoblade'}
 for univ in universes:
     print('\t\t' + univ)
-    sys.stdout.flush
-    if univ == 'wowwiki' or univ == "enmarveldatabase" or univ == "starwars" or univ == "harrypotter":
+    sys.stdout.flush()
+    if univ in big_universes:
         continue;
     occurrenceFile = basedir + univ + "/type-occurrence"
     occurrences = {}