diff --git a/vrobbler/apps/videos/models.py b/vrobbler/apps/videos/models.py index d6f5251..c14295b 100644 --- a/vrobbler/apps/videos/models.py +++ b/vrobbler/apps/videos/models.py @@ -306,7 +306,9 @@ class Video(ScrobblableMixin): return video @classmethod - def get_from_imdb_id(cls, imdb_id: str, overwrite: bool = False): + def get_from_imdb_id( + cls, imdb_id: str, overwrite: bool = False + ) -> "Video": video, created = cls.objects.get_or_create(imdb_id=imdb_id) if not created and not overwrite: return video