Skip to content

Commit

Permalink
fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix committed Mar 19, 2024
1 parent e757ac2 commit 0a22e7f
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions examples/neural_compressor/test_examples.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ def test_run_glue(self):

with patch.object(sys, "argv", test_args):
run_glue.main()
results = get_results(tmp_dir)
get_results(tmp_dir)

def test_run_qa(self):
with tempfile.TemporaryDirectory() as tmp_dir:
Expand All @@ -103,7 +103,7 @@ def test_run_qa(self):

with patch.object(sys, "argv", test_args):
run_qa.main()
results = get_results(tmp_dir)
get_results(tmp_dir)

def test_run_ner(self):
with tempfile.TemporaryDirectory() as tmp_dir:
Expand All @@ -128,7 +128,7 @@ def test_run_ner(self):

with patch.object(sys, "argv", test_args):
run_ner.main()
results = get_results(tmp_dir)
get_results(tmp_dir)

def test_run_swag(self):
with tempfile.TemporaryDirectory() as tmp_dir:
Expand All @@ -152,7 +152,7 @@ def test_run_swag(self):

with patch.object(sys, "argv", test_args):
run_swag.main()
results = get_results(tmp_dir)
get_results(tmp_dir)

def test_run_clm(self):
quantization_approach = "dynamic"
Expand Down Expand Up @@ -181,7 +181,7 @@ def test_run_clm(self):

with patch.object(sys, "argv", test_args):
run_clm.main()
results = get_results(tmp_dir)
get_results(tmp_dir)

def test_run_mlm(self):
quantization_approach = "static"
Expand Down Expand Up @@ -210,7 +210,7 @@ def test_run_mlm(self):

with patch.object(sys, "argv", test_args):
run_mlm.main()
results = get_results(tmp_dir)
get_results(tmp_dir)


if __name__ == "__main__":
Expand Down

0 comments on commit 0a22e7f

Please sign in to comment.