Skip to content

Commit 8679ed3

Browse files
committed
fix: greptile comments (#129)
1 parent e46e5f6 commit 8679ed3

1 file changed

Lines changed: 19 additions & 18 deletions

File tree

lib/classifier/cli.rb

Lines changed: 19 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -392,7 +392,7 @@ def list_remote_models
392392
type = info['type'] || 'unknown'
393393
size = info['size'] || 'unknown'
394394
desc = info['description'] || ''
395-
categories = info['categories'].map(&:downcase).join(', ') || ''
395+
categories = (info['categories'] || []).map(&:downcase).join(', ')
396396
version = info['version'] || ''
397397
author = info['author'] || ''
398398
@output << format(
@@ -401,13 +401,14 @@ def list_remote_models
401401
name: name, desc: desc.slice(0, 40), type: type, categories: categories,
402402
version: version, author: author, size: size
403403
)
404-
else
405-
index['models'].each do |name, info|
406-
type = info['type'] || 'unknown'
407-
size = info['size'] || 'unknown'
408-
desc = info['description'] || ''
409-
@output << format('%-20<name>s %<desc>s (%<type>s, %<size>s)', name: name, desc: desc.slice(0, 40), type: type, size: size)
410-
end
404+
return
405+
end
406+
407+
index['models'].each do |name, info|
408+
type = info['type'] || 'unknown'
409+
size = info['size'] || 'unknown'
410+
desc = info['description'] || ''
411+
@output << format('%-20<name>s %<desc>s (%<type>s, %<size>s)', name: name, desc: desc.slice(0, 40), type: type, size: size)
411412
end
412413
end
413414

@@ -452,22 +453,23 @@ def list_local_models
452453
info = load_model_info(model[:path])
453454
type = info['type'] || 'unknown'
454455
version = info['version']
455-
categories = info['categories'].keys.map(&:downcase).join(', ')
456+
categories = (info['categories'] || {}).keys.map(&:downcase).join(', ')
456457
size = File.size(model[:path])
457458
@output << format(
458459
"Name: %<name>s\nType: %<type>s\n" \
459460
"Categories: %<categories>s\nVersion: %<version>s\nSize: %<size>s",
460461
name: display_name, type: type, categories: categories,
461462
version: version, size: human_size(size)
462463
)
463-
else
464-
models.each do |model|
465-
info = load_model_info(model[:path])
466-
type = info['type'] || 'unknown'
467-
display_name = model[:registry] ? "@#{model[:registry]}:#{model[:name]}" : model[:name]
468-
size = File.size(model[:path])
469-
@output << format('%-30<name>s (%<type>s, %<size>s)', name: display_name, type: type, size: human_size(size))
470-
end
464+
return
465+
end
466+
467+
models.each do |model|
468+
info = load_model_info(model[:path])
469+
type = info['type'] || 'unknown'
470+
display_name = model[:registry] ? "@#{model[:registry]}:#{model[:name]}" : model[:name]
471+
size = File.size(model[:path])
472+
@output << format('%-30<name>s (%<type>s, %<size>s)', name: display_name, type: type, size: human_size(size))
471473
end
472474
end
473475

@@ -844,7 +846,6 @@ def show_getting_started
844846
@output << 'Run "classifier --help" for full usage.'
845847
end
846848

847-
# Detect registry and model
848849
# @rbs (String?) -> [String?, String?]
849850
def detect_registry_and_model(arg)
850851
return nil, nil if arg.nil?

0 commit comments

Comments
 (0)