Better docs

django_ltree/managers.py

@@ -9,21 +9,21 @@
 
 
 class TreeManager(models.Manager):
-    def get_queryset(self):
+    def get_queryset(self) -> TreeQuerySet["TreeModel"]:
         """Returns a queryset with the models ordered by `path`"""
         return TreeQuerySet(model=self.model, using=self._db).order_by("path")
 
-    def roots(self) -> models.QuerySet["TreeModel"]:
+    def roots(self) -> TreeQuerySet["TreeModel"]:
         """Returns the roots of a given model"""
         return self.filter().roots()
 
-    def children(self, path: str) -> models.QuerySet["TreeModel"]:
+    def children(self, path: str) -> TreeQuerySet["TreeModel"]:
         """Returns the childrens of a given object"""
         return self.filter().children(path)
 
     def create_child(
         self, parent: "TreeModel" = None, **kwargs
-    ) -> models.QuerySet["TreeModel"]:
+    ) -> TreeQuerySet["TreeModel"]:
         """Creates a tree child with or without parent"""
         paths_in_use = parent.children() if parent else self.roots()
         prefix = parent.path if parent else None