tc

pytorch · Sep 13, 2024 · 039dca8 · 039dca8
1 parent f1cbba1
commit 039dca8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tools/torchci/clickhouse.py b/tools/torchci/clickhouse.py
@@ -9,8 +9,11 @@
 
 @lru_cache(maxsize=1)
 def get_clickhouse_client() -> Any:
+    endpoint = os.environ["CLICKHOUSE_ENDPOINT"]
+    if endpoint.startswith("https://"):
+        endpoint = endpoint[len("https://") :]
     return clickhouse_connect.get_client(
-        host=os.environ["CLICKHOUSE_ENDPOINT"],
+        host=endpoint,
         user=os.environ["CLICKHOUSE_USERNAME"],
         password=os.environ["CLICKHOUSE_PASSWORD"],
         secure=True,