summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--python/fatcat_tools/transforms.py3
-rw-r--r--python/fatcat_tools/workers/elastic.py2
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