diff --git a/examples/cpu/features/int8_recipe_tuning/int8_autotune.py b/examples/cpu/features/int8_recipe_tuning/int8_autotune.py index b70155ca4..05b0a3d10 100644 --- a/examples/cpu/features/int8_recipe_tuning/int8_autotune.py +++ b/examples/cpu/features/int8_recipe_tuning/int8_autotune.py @@ -84,7 +84,7 @@ def train(dataloader, model, loss_fn, optimizer): epochs = 5 for t in range(epochs): - print(f"Epoch {t+1}\n-------------------------------") + print(f"Epoch {t + 1}\n-------------------------------") train(train_dataloader, model, loss_fn, optimizer) print("Done!") diff --git a/examples/cpu/llm/inference/distributed/run_generation_with_deepspeed.py b/examples/cpu/llm/inference/distributed/run_generation_with_deepspeed.py index ecf8eea86..ccad56982 100644 --- a/examples/cpu/llm/inference/distributed/run_generation_with_deepspeed.py +++ b/examples/cpu/llm/inference/distributed/run_generation_with_deepspeed.py @@ -1017,7 +1017,7 @@ def trace_handler(prof): generated, _ = generate() t_generate_span = time.time() - t_generate_start for i, o, _ in generated: - print_rank0(f"{'-'*60}\nin={i}\nout={o}\n") + print_rank0(f"{'-' * 60}\nin={i}\nout={o}\n") # benchmark it! else: diff --git a/intel_extension_for_pytorch/__init__.py b/intel_extension_for_pytorch/__init__.py index 826562117..f1d577d91 100644 --- a/intel_extension_for_pytorch/__init__.py +++ b/intel_extension_for_pytorch/__init__.py @@ -1,6 +1,6 @@ # This Python file uses the following encoding: utf-8 import re -import os +import sys import torch @@ -33,7 +33,7 @@ + f"{ipex_version}.*, but PyTorch {torch.__version__} is found. " + "Please switch to the matching version and run again." ) - os.exit(127) + sys.exit(127) import os