Removed MOFKA_PROTOCOL from mofka pub/sub initialization

globus-labs · Dec 11, 2024 · e9180a4 · e9180a4
1 parent 32ee2aa
commit e9180a4
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/mofa/proxyqueue.py b/mofa/proxyqueue.py
@@ -40,7 +40,7 @@
     from proxystore.ex.stream.shims.mofka import MofkaPublisher
 
     assert (MOFKA_GROUPFILE := os.environ["MOFKA_GROUPFILE"])
-    assert (MOFKA_PROTOCOL := os.environ["MOFKA_PROTOCOL"])
+    assert os.environ["MOFKA_PROTOCOL"]
 
 
 def oauth_cb(oauth_config):
@@ -124,9 +124,7 @@ def connect_request_producer(self):
                     stores={k: self.store for k in self.proxy_topics},
                 )
         else:
-            publisher = MofkaPublisher(
-                protocol=MOFKA_PROTOCOL, group_file=MOFKA_GROUPFILE
-            )
+            publisher = MofkaPublisher(group_file=MOFKA_GROUPFILE)
             self.request_producer = StreamProducer(publisher=publisher)
 
     def connect_request_consumer(self):
@@ -143,7 +141,6 @@ def connect_request_consumer(self):
                 subscriber = KafkaSubscriber(client=consumer)
             else:
                 subscriber = MofkaSubscriber(
-                    protocol=MOFKA_PROTOCOL,
                     group_file=MOFKA_GROUPFILE,
                     topic_name=request_topic,
                     subscriber_name=str(f"MOFA-request-{uuid4()}"),
@@ -166,7 +163,6 @@ def connect_result_consumer(self, topic):
                 subscriber = KafkaSubscriber(client=consumer)
             else:
                 subscriber = MofkaSubscriber(
-                    protocol=MOFKA_PROTOCOL,
                     group_file=MOFKA_GROUPFILE,
                     topic_name=topic,
                     subscriber_name=str(f"MOFA-result-{uuid4()}"),