fix syntax bugs (container elastic worker)

2 files changed, 6 insertions, 6 deletions

diff --git a/python/fatcat_tools/workers/elasticsearch.py b/python/fatcat_tools/workers/elasticsearch.py
index 6e336e52..fc96e3d4 100644
--- a/python/fatcat_tools/workers/elasticsearch.py
+++ b/python/fatcat_tools/workers/elasticsearch.py
@@ -4,7 +4,7 @@ import time
 import requests
 from pykafka.common import OffsetType
 
-from fatcat_client import ReleaseEntity, ApiClient
+from fatcat_client import ReleaseEntity, ContainerEntity, ApiClient
 from fatcat_tools import *
 from .worker_common import FatcatWorker
 
@@ -62,10 +62,10 @@ class ElasticsearchContainerWorker(ElasticsearchReleaseWorker):
             elasticsearch_backend="http://localhost:9200", elasticsearch_index="fatcat"):
         super().__init__(kafka_hosts=kafka_hosts,
                          consume_topic=consume_topic,
-                         poll_interval,
-                         offset,
-                         elasticsearch_backend,
-                         elasticsearch_index)
+                         poll_interval=poll_interval,
+                         offset=offset,
+                         elasticsearch_backend=elasticsearch_backend,
+                         elasticsearch_index=elasticsearch_index)
         self.entity_type = ContainerEntity
         self.elasticsearch_document_name = "container"
         self.transform_func = container_to_elasticsearch
diff --git a/python/fatcat_worker.py b/python/fatcat_worker.py
index 4455d29e..b9e5b8fd 100755
--- a/python/fatcat_worker.py
+++ b/python/fatcat_worker.py
@@ -6,7 +6,7 @@ import datetime
 import raven
 
 from fatcat_tools import public_api
-from fatcat_tools.workers import ChangelogWorker, EntityUpdatesWorker, ElasticsearchReleaseWorker
+from fatcat_tools.workers import ChangelogWorker, EntityUpdatesWorker, ElasticsearchReleaseWorker, ElasticsearchContainerWorker
 
 # Yep, a global. Gets DSN from `SENTRY_DSN` environment variable
 sentry_client = raven.Client()