File tree Expand file tree Collapse file tree 2 files changed +9
-8
lines changed Expand file tree Collapse file tree 2 files changed +9
-8
lines changed Original file line number Diff line number Diff line change @@ -190,27 +190,28 @@ def get_free_port():
190
190
except Exception as e :
191
191
logger .error (f"Error handling existing container: { str (e )} " )
192
192
193
- model_name = next (
194
- name for name , cfg in TEST_CONFIGS .items () if cfg ["model_id" ] == model_id
195
- )
196
-
197
- tgi_args = TEST_CONFIGS [model_name ]["args" ].copy ()
193
+ tgi_args = TEST_CONFIGS [test_name ]["args" ].copy ()
198
194
199
195
env = BASE_ENV .copy ()
200
196
201
197
# Add model_id to env
202
198
env ["MODEL_ID" ] = model_id
203
199
204
200
# Add env config that is definied in the fixture parameter
205
- if "env_config" in TEST_CONFIGS [model_name ]:
206
- env .update (TEST_CONFIGS [model_name ]["env_config" ].copy ())
201
+ if "env_config" in TEST_CONFIGS [test_name ]:
202
+ env .update (TEST_CONFIGS [test_name ]["env_config" ].copy ())
207
203
208
204
volumes = [f"{ DOCKER_VOLUME } :/data" ]
209
205
logger .debug (f"Using volume { volumes } " )
210
206
211
207
try :
208
+ logger .debug (f"Using command { tgi_args } " )
212
209
logger .info (f"Creating container with name { container_name } " )
213
210
211
+ logger .debug (f"Using environment { env } " )
212
+ logger .debug (f"Using volumes { volumes } " )
213
+ logger .debug (f"HABANA_RUN_ARGS { HABANA_RUN_ARGS } " )
214
+
214
215
# Log equivalent docker run command for debugging, this is not actually executed
215
216
container = client .containers .run (
216
217
DOCKER_IMAGE ,
Original file line number Diff line number Diff line change 3
3
from typing import Dict , Any , Generator
4
4
5
5
import pytest
6
- from test_generate import TEST_CONFIGS
6
+ from test_gaudi_generate import TEST_CONFIGS
7
7
8
8
UNKNOWN_CONFIGS = {
9
9
name : config
You can’t perform that action at this time.
0 commit comments