From 0a83392cbf73d5049dd7da1446102a430536d83a Mon Sep 17 00:00:00 2001 From: Cristian Date: Mon, 14 Sep 2020 13:27:33 -0500 Subject: [PATCH] fix: Replace `any` typing with Union[Iterable[Link], QuerySet] in `archive_links` --- archivebox/extractors/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/archivebox/extractors/__init__.py b/archivebox/extractors/__init__.py index cc310570..6c277809 100644 --- a/archivebox/extractors/__init__.py +++ b/archivebox/extractors/__init__.py @@ -2,7 +2,7 @@ __package__ = 'archivebox.extractors' import os -from typing import Optional, List, Iterable +from typing import Optional, List, Iterable, Union, TYPE_CHECKING from datetime import datetime from django.db.models import QuerySet @@ -130,7 +130,7 @@ def archive_link(link: Link, overwrite: bool=False, methods: Optional[Iterable[s return link @enforce_types -def archive_links(all_links: any, overwrite: bool=False, methods: Optional[Iterable[str]]=None, out_dir: Optional[str]=None) -> List[Link]: +def archive_links(all_links: Union[Iterable[Link], QuerySet], overwrite: bool=False, methods: Optional[Iterable[str]]=None, out_dir: Optional[str]=None) -> List[Link]: if type(all_links) is QuerySet: num_links: int = all_links.count()