return type

cajal · Jul 10, 2023 · bac78c8 · bac78c8
1 parent 8ce328b
commit bac78c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/djutils/links.py b/djutils/links.py
@@ -44,7 +44,7 @@ def clean(cls):
         for table in cls.tables:
             part = getattr(cls, table)
             master = cls & {f"{cls.name}_type": part.__name__}
-            keys += (master - part).fetch("KEY")
+            keys += (master - part).fetch(dj.key)
 
         (cls & keys).delete()
 
@@ -70,7 +70,7 @@ def query(cls, link_type, link_key=None):
 
         Returns
         -------
-        dj.Lookup
+        djutils.links.Link
             link table restricted by link type, and optionally restricted by link key
         """
         keys = cls & {f"{cls.name}_type": link_type}