@@ -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\n Type: %<type>s\n " \
459460 "Categories: %<categories>s\n Version: %<version>s\n Size: %<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