From 8c3fb7409d4438c04d1dd41ee07b25e5897ae90a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Elan=20Ruusam=C3=A4e?= Date: Fri, 16 Apr 2021 21:42:31 +0300 Subject: [PATCH] Add typing to fool IDE --- tests/test_tv_lookup.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_tv_lookup.py b/tests/test_tv_lookup.py index b0cfbe4b22..e2159963d6 100755 --- a/tests/test_tv_lookup.py +++ b/tests/test_tv_lookup.py @@ -1,10 +1,13 @@ #!/usr/bin/env python3 -m pytest +from typing import Union + +from trakt.tv import TVShow from plex_trakt_sync.plex_api import PlexLibraryItem from plex_trakt_sync.trakt_api import TraktApi -def make(cls=None, **kwargs): +def make(cls=None, **kwargs) -> Union[TVShow]: cls = cls if cls is not None else "object" # https://stackoverflow.com/a/2827726/2314626 return type(cls, (object,), kwargs)