diff options
-rw-r--r-- | python/fatcat_tools/transforms.py | 3 | ||||
-rw-r--r-- | python/fatcat_tools/workers/elastic.py | 2 |
2 files changed, 2 insertions, 3 deletions
diff --git a/python/fatcat_tools/transforms.py b/python/fatcat_tools/transforms.py index c454536b..ecaf3366 100644 --- a/python/fatcat_tools/transforms.py +++ b/python/fatcat_tools/transforms.py @@ -1,7 +1,6 @@ import collections -from fatcat_client.models import ReleaseEntity -from fatcat_client.api_client import ApiClient +from fatcat_client import ReleaseEntity, ApiClient def entity_to_json(entity): ac = ApiClient() diff --git a/python/fatcat_tools/workers/elastic.py b/python/fatcat_tools/workers/elastic.py index 46632792..22ec6fa6 100644 --- a/python/fatcat_tools/workers/elastic.py +++ b/python/fatcat_tools/workers/elastic.py @@ -3,7 +3,7 @@ import json import time import requests from fatcat_tools.workers.worker_common import FatcatWorker -from fatcat_client.models import ReleaseEntity +from fatcat_client import ReleaseEntity from fatcat_tools.transforms import * from pykafka.common import OffsetType |