From a4a3142a981e7e9ca220e935ebae8a0dc2d9331c Mon Sep 17 00:00:00 2001
From: changwangss <chang1.wang@intel.com>
Date: Mon, 25 Sep 2023 19:00:22 -0700
Subject: [PATCH] fix style

Signed-off-by: changwangss <chang1.wang@intel.com>
---
 tests/ipex/test_inference.py | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/tests/ipex/test_inference.py b/tests/ipex/test_inference.py
index f1aa44468f..d21267fbdb 100644
--- a/tests/ipex/test_inference.py
+++ b/tests/ipex/test_inference.py
@@ -15,7 +15,6 @@
 import unittest
 
 import torch
-from parameterized import parameterized
 
 # TODO : add more tasks
 from transformers import (
@@ -29,6 +28,7 @@
 
 from optimum.intel import inference_mode as ipex_inference_mode
 from optimum.intel.generation.modeling import TSModelForCausalLM
+from parameterized import parameterized
 
 
 MODEL_NAMES = {
@@ -56,12 +56,7 @@ class IPEXIntegrationTest(unittest.TestCase):
         "roberta",
     )
 
-    TEXT_GENERATION_SUPPORTED_ARCHITECTURES = (
-        "gptj",
-        "gpt2",
-        "gpt_neo",
-        "gpt_bigcode"
-    )
+    TEXT_GENERATION_SUPPORTED_ARCHITECTURES = ("gptj", "gpt2", "gpt_neo", "gpt_bigcode")
 
     QA_SUPPORTED_ARCHITECTURES = (
         "bert",