Skip to content

Commit a2595d7

Browse files
committed
*.py: logging basiconfig refactor to use conditional expression
1 parent 27d97d8 commit a2595d7

File tree

7 files changed

+7
-29
lines changed

7 files changed

+7
-29
lines changed

convert-persimmon-to-gguf.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -73,10 +73,7 @@ def main():
7373
parser.add_argument("--adept-inference-dir", type=str, help="path to adept-inference code directory")
7474
parser.add_argument("--verbose", action="store_true", help="increase output verbosity")
7575
args = parser.parse_args()
76-
if args.verbose:
77-
logging.basicConfig(level=logging.DEBUG)
78-
else:
79-
logging.basicConfig(level=logging.INFO)
76+
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
8077
sys.path.append(str(args.adept_inference_dir))
8178
persimmon_model = torch.load(args.ckpt_path)
8279
hparams = persimmon_model['args']

ggml_vk_generate_shaders.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2775,10 +2775,7 @@ async def withSemaphore(sem, task):
27752775

27762776
args = parser.parse_args()
27772777

2778-
if args.verbose:
2779-
logging.basicConfig(level=logging.DEBUG)
2780-
else:
2781-
logging.basicConfig(level=logging.INFO)
2778+
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
27822779

27832780
if args.glslc:
27842781
GLSLC = args.glslc

gguf-py/scripts/gguf-convert-endian.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -110,10 +110,7 @@ def main() -> None:
110110

111111
args = parser.parse_args(None if len(sys.argv) > 1 else ["--help"])
112112

113-
if args.verbose:
114-
logging.basicConfig(level=logging.DEBUG)
115-
else:
116-
logging.basicConfig(level=logging.INFO)
113+
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
117114

118115
logger.info(f'* Loading: {args.model}')
119116
reader = gguf.GGUFReader(args.model, 'r' if args.dry_run else 'r+')

gguf-py/scripts/gguf-dump.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -111,10 +111,7 @@ def main() -> None:
111111

112112
args = parser.parse_args(None if len(sys.argv) > 1 else ["--help"])
113113

114-
if args.verbose:
115-
logging.basicConfig(level=logging.DEBUG)
116-
else:
117-
logging.basicConfig(level=logging.INFO)
114+
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
118115

119116
if not args.json:
120117
logger.info(f'* Loading: {args.model}')

gguf-py/scripts/gguf-set-metadata.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -84,10 +84,7 @@ def main() -> None:
8484

8585
args = parser.parse_args(None if len(sys.argv) > 1 else ["--help"])
8686

87-
if args.verbose:
88-
logging.basicConfig(level=logging.DEBUG)
89-
else:
90-
logging.basicConfig(level=logging.INFO)
87+
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
9188

9289
logger.info(f'* Loading: {args.model}')
9390
reader = GGUFReader(args.model, 'r' if args.dry_run else 'r+')

tests/test-tokenizer-0-falcon.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,7 @@
1313
parser.add_argument("--verbose", action="store_true", help="increase output verbosity")
1414

1515
args = parser.parse_args()
16-
if args.verbose:
17-
logging.basicConfig(level=logging.DEBUG)
18-
else:
19-
logging.basicConfig(level=logging.INFO)
20-
16+
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
2117
dir_tokenizer = args.dir_tokenizer
2218

2319
tokenizer = AutoTokenizer.from_pretrained(dir_tokenizer)

tests/test-tokenizer-0-llama.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,7 @@
1414

1515
args = parser.parse_args()
1616

17-
if args.verbose:
18-
logging.basicConfig(level=logging.DEBUG)
19-
else:
20-
logging.basicConfig(level=logging.INFO)
17+
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
2118

2219
dir_tokenizer = args.dir_tokenizer
2320

0 commit comments

Comments
 (0)