diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py index 354f04f5..3130421e 100644 --- a/integration-tests/conftest.py +++ b/integration-tests/conftest.py @@ -589,7 +589,6 @@ def generate_multi(): max_new_tokens: int, seed: Optional[int] = None, ) -> List[Response]: - import numpy as np arange = np.arange(len(prompts))