Skip to content

Commit b8f5d41

Browse files
committed
fix code style
1 parent 2102c25 commit b8f5d41

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/ipex/test_modeling.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -228,7 +228,9 @@ def test_compare_to_transformers(self, model_arch):
228228
self.assertTrue(ipex_model.use_cache)
229229
tokenizer = AutoTokenizer.from_pretrained(model_id)
230230
tokens = tokenizer(
231-
"This is a sample", return_tensors="pt", return_token_type_ids=False if model_arch in ("llama", "llama2") else None
231+
"This is a sample",
232+
return_tensors="pt",
233+
return_token_type_ids=False if model_arch in ("llama", "llama2") else None,
232234
)
233235
position_ids = None
234236
if model_arch.replace("_", "-") in MODEL_TYPES_REQUIRING_POSITION_IDS:

0 commit comments

Comments
 (0)