fix: types

dhdaines · Jan 22, 2025 · 6a31e43 · 6a31e43
1 parent 86ca051
commit 6a31e43
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/playa/document.py b/playa/document.py
@@ -1406,8 +1406,10 @@ def __init__(
     ) -> None:
         self.docref = _ref_document(doc)
         if pages is not None:
-            self._pages = pages
-            self._labels: Dict[str, Page] = {page.label: page for page in pages}
+            self._pages = list(pages)
+            self._labels: Dict[str, Page] = {
+                page.label: page for page in pages if page.label is not None
+            }
         else:
             self._init_pages(doc)
 

diff --git a/playa/worker.py b/playa/worker.py
@@ -9,7 +9,7 @@
     from playa.page import Page
 
 # Type signature of document reference
-DocumentRef = Union[weakref.ReferenceType["Document"], int]
+DocumentRef = int
 # Type signature of page reference
 PageRef = Tuple[DocumentRef, int]