Fix some issue with relation updates

jmccrae · Jan 17, 2025 · 23ab580 · 23ab580
1 parent 43be7bb
commit 23ab580
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/src/automaton.rs b/src/automaton.rs
@@ -126,14 +126,14 @@ pub fn apply_automaton(actions : Vec<Action>, wn : &mut Lexicon,
                         change_manager::insert_sense_relation(wn, 
                             sense.clone(), 
                             SenseRelType::from(&relation)
-                                .ok_or(format!("Bad relation {}", relation))?,
+                                .ok_or(format!("Bad relation {}.", relation))?,
                             target_sense, changes);
                     },
                     None => {
                         change_manager::insert_rel(wn, 
                             &source,
                             &SynsetRelType::from(&relation)
-                                .ok_or(format!("Bad relation {}", relation))?,
+                                .ok_or(format!("Bad relation {}.", relation))?,
                             &target, changes);
                     }
                 }

diff --git a/src/rels.rs b/src/rels.rs
@@ -514,7 +514,9 @@ impl SenseRelType {
             "domain_region" => Some(SenseRelType::DomainRegion),
             "has_domain_region" => Some(SenseRelType::HasDomainRegion),
             "exemplifies" => Some(SenseRelType::Exemplifies),
+            "exemplifies_sense" => Some(SenseRelType::Exemplifies),
             "is_exemplified_by" => Some(SenseRelType::IsExemplifiedBy),
+            "is_exemplified_by_sense" => Some(SenseRelType::IsExemplifiedBy),
             "similar" => Some(SenseRelType::Similar),
             "other" => Some(SenseRelType::Other),
             "agent" => Some(SenseRelType::Agent),
@@ -531,7 +533,9 @@ impl SenseRelType {
             "destination" => Some(SenseRelType::Destination),
             "body_part" => Some(SenseRelType::BodyPart),
             "vehicle" => Some(SenseRelType::Vehicle),
-            _ => None
+            _ => {
+                None
+            }
         }
     }
 

diff --git a/src/wordnet.rs b/src/wordnet.rs
@@ -476,8 +476,8 @@ impl Lexicon {
                 None => {}
             }
         } else {
-            match self.synset_by_id_mut(source) {
-                Some(ss) => ss.insert_rel(&rel, target),
+            match self.synset_by_id_mut(target) {
+                Some(ss) => ss.insert_rel(&rel, source),
                 None => {}
             }
         }