Skip to content

Commit 39f005e

Browse files
committed
fix precommit issues
1 parent 72982a5 commit 39f005e

File tree

4 files changed

+5
-8
lines changed

4 files changed

+5
-8
lines changed

integration_tests/Makefile

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,3 @@ run-integration-tests-hpu:
22
DOCKER_VOLUME=/home/sdp/git/text-embeddings-inference/data \
33
HF_TOKEN=`cat ${HOME}/.cache/huggingface/token` \
44
uv run pytest --durations=0 -sv .
5-
6-

integration_tests/README.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,3 @@ Then you can run the tests.
1515
```bash
1616
make -C integration_tests run-integration-tests-hpu
1717
```
18-

integration_tests/gaudi/conftest.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -179,14 +179,14 @@ def docker_launcher(
179179
logger.error(f"Error handling existing container: {str(e)}")
180180

181181
tei_args = TEST_CONFIGS[test_name]["args"].copy()
182-
182+
183183
# add model_id to tei args
184184
tei_args.append("--model-id")
185185
tei_args.append(model_id)
186186

187187
env = BASE_ENV.copy()
188188
env["HF_TOKEN"] = os.getenv("HF_TOKEN")
189-
189+
190190
# Add env config that is definied in the fixture parameter
191191
if "env_config" in TEST_CONFIGS[test_name]:
192192
env.update(TEST_CONFIGS[test_name]["env_config"].copy())
@@ -255,4 +255,4 @@ def docker_launcher(
255255
except Exception as e:
256256
logger.warning(f"Error cleaning up container: {str(e)}")
257257

258-
return docker_launcher
258+
return docker_launcher

integration_tests/gaudi/test_embed.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,8 @@ async def test_model_single_request(
7272
response = await tei_client.generate(
7373
input,
7474
)
75-
75+
7676
response_array = np.array(response)
7777
expected_array = np.array(eval(expected_outputs["expected_output"]))
7878

79-
assert np.allclose(response_array, expected_array, rtol=1e-5, atol=1e-5)
79+
assert np.allclose(response_array, expected_array, rtol=1e-5, atol=1e-5)

0 commit comments

Comments
 (0)