bug fix: documentation building in CI

airtai · Dec 25, 2022 · a14a046 · a14a046
1 parent 7cb05ca
commit a14a046
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/fast_kafka_api/testing.py b/fast_kafka_api/testing.py
@@ -25,8 +25,10 @@
 logger = get_logger(__name__)
 
 # %% ../nbs/999_Test_Utils.ipynb 5
-kafka_server_url = environ["KAFKA_HOSTNAME"]
-kafka_server_port = environ["KAFKA_PORT"]
+kafka_server_url = (
+    environ["KAFKA_HOSTNAME"] if "KAFKA_HOSTNAME" in environ else "localhost"
+)
+kafka_server_port = environ["KAFKA_PORT"] if "KAFKA_PORT" in environ else "9092"
 
 kafka_config = {
     "bootstrap.servers": f"{kafka_server_url}:{kafka_server_port}",

diff --git a/nbs/999_Test_Utils.ipynb b/nbs/999_Test_Utils.ipynb
@@ -81,12 +81,14 @@
    "source": [
     "# | export\n",
     "\n",
-    "kafka_server_url = environ[\"KAFKA_HOSTNAME\"]\n",
-    "kafka_server_port = environ[\"KAFKA_PORT\"]\n",
+    "kafka_server_url = (\n",
+    "    environ[\"KAFKA_HOSTNAME\"] if \"KAFKA_HOSTNAME\" in environ else \"localhost\"\n",
+    ")\n",
+    "kafka_server_port = environ[\"KAFKA_PORT\"] if \"KAFKA_PORT\" in environ else \"9092\"\n",
     "\n",
     "kafka_config = {\n",
     "    \"bootstrap.servers\": f\"{kafka_server_url}:{kafka_server_port}\",\n",
-    "    #\"group.id\": f\"{kafka_server_url}:{kafka_server_port}_group\"\n",
+    "    # \"group.id\": f\"{kafka_server_url}:{kafka_server_port}_group\"\n",
     "}"
    ]
   },