phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test: fix linting testing

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-19 18:40:33 -07:00
parent 4198901a2d
commit 04f8840a92
3 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_router.py
View file

@ -43,7 +43,7 @@ model_list = [{ # list of model deployments
"rpm": 9000
}]
router = Router(model_list=model_list, redis_host=os.getenv("REDIS_HOST"), redis_password=os.getenv("REDIS_PASSWORD"), redis_port=os.getenv("REDIS_PORT"))
router = Router(model_list=model_list, redis_host=os.getenv("REDIS_HOST"), redis_password=os.getenv("REDIS_PASSWORD"), redis_port=int(os.getenv("REDIS_PORT"))) # type: ignore
completions = []
with ThreadPoolExecutor(max_workers=100) as executor:
@ -52,7 +52,7 @@ with ThreadPoolExecutor(max_workers=100) as executor:
"messages": [{"role": "user", "content": "Hey, how's it going?"}]
}
for _ in range(20):
future = executor.submit(router.completion, **kwargs)
future = executor.submit(router.completion, **kwargs) # type: ignore
completions.append(future)
# Retrieve the results from the futures