save

databricks · Feb 14, 2024 · 5e64f2e · 5e64f2e
1 parent 733a3c6
commit 5e64f2e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/lilac/data/cluster_titling.py b/lilac/data/cluster_titling.py
@@ -14,6 +14,7 @@
 from tenacity import retry, retry_if_exception_type, stop_after_attempt, wait_random_exponential
 
 from ..batch_utils import group_by_sorted_key_iter
+from ..env import env
 from ..schema import (
   Item,
 )
@@ -24,7 +25,6 @@
 )
 from ..tasks import TaskInfo
 from ..utils import chunks, log
-from ..env import env
 
 _TOP_K_CENTRAL_DOCS = 7
 _TOP_K_CENTRAL_TITLES = 20

diff --git a/lilac/server_mount_test.py b/lilac/server_mount_test.py
@@ -11,7 +11,7 @@
 
 
 @app.get('/')
-def read_main():
+def read_main() -> dict:
   """The main endpoint."""
   return {'message': 'hello world'}
 
@@ -25,9 +25,6 @@ def test_mount_root() -> None:
   assert response.status_code == 200
   assert response.json() == {'message': 'hello world'}
 
-  response = client.get(f'{MOUNT_POINT}/', allow_redirects=False)
-  assert response.status_code == 200
-
 
 def test_mount_slash_redirect() -> None:
   response = client.get(f'{MOUNT_POINT}/auth_info/', allow_redirects=False)