Adjust import style

This commit is contained in:
Edward Betts 2023-11-09 07:39:39 +01:00
parent e7004a08f4
commit d814ecf3a4

View file

@ -1,6 +1,6 @@
"""Class to represent artwork."""
from . import mediawiki
from .mediawiki import get_entity, get_entity_with_cache
from .type import Claims, Entity, Sitelinks
@ -12,7 +12,7 @@ class Artwork:
def __init__(self, qid: str) -> None:
"""Init."""
entity = mediawiki.get_entity_with_cache(qid)
entity = get_entity_with_cache(qid)
assert entity
self.entity = entity
self.item_id = int(qid[1:])
@ -48,7 +48,7 @@ class Artwork:
self.artist_entities = []
for artist in self.artists_claim:
artist_entity = mediawiki.get_entity(artist["id"])
artist_entity = get_entity(artist["id"])
assert artist_entity
self.artist_entities.append(artist_entity)