add location to all themes (including experimental

phueb · Apr 9, 2022 · e0f9292 · e0f9292
1 parent 9994d3a
commit e0f9292
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/missingadjunct/corpus.py b/missingadjunct/corpus.py
@@ -109,7 +109,7 @@ def get_logical_forms(self) -> Generator[LogicalForm, None, None]:
 
                 if lf.theme in self.experimental_themes:
                     lf.instrument = None
-                    lf.location = None
+                    # lf.location = None  # the point of location is to cluster all themes (also experimental themes)
 
                 if not self.include_location:
                     lf.location = None
@@ -118,7 +118,7 @@ def get_logical_forms(self) -> Generator[LogicalForm, None, None]:
                     yield lf
 
         # for remaining epochs, sample randomly from agent and theme
-        epoch = 0
+        epoch = -1
         for theme_class, agent_class, verb in self._gen_templates():
             epoch += 1
 
@@ -144,7 +144,7 @@ def get_logical_forms(self) -> Generator[LogicalForm, None, None]:
 
             if lf.theme in self.experimental_themes:
                 lf.instrument = None
-                lf.location = None
+                # lf.location = None  # the point of location is to cluster all themes (including experimental themes)
 
             if not self.include_location:
                 lf.location = None