Implement get_refs function for DrtLambdaExpression (#2847)

* Implement get_refs function for DrtLambdaExpression * Update format Co-authored-by: Ahmet Yildirim <ahmet.yildirim@iln.uio.no>
nltk · Oct 9, 2021 · 6d39c90 · 6d39c90
1 parent 3dcfeb1
commit 6d39c90
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/nltk/sem/drt.py b/nltk/sem/drt.py
@@ -717,6 +717,12 @@ def _pretty(self):
             + ["    " + blank + line for line in term_lines[3:]]
         )
 
+    def get_refs(self, recursive=False):
+        """:see: AbstractExpression.get_refs()"""
+        return (
+            [self.variable] + self.term.get_refs(True) if recursive else [self.variable]
+        )
+
 
 class DrtBinaryExpression(DrtExpression, BinaryExpression):
     def get_refs(self, recursive=False):