@@ -503,12 +503,14 @@ func TestAppBrickInstanceModelsDetails(t *testing.T) {
503503 {Name : "EI_OBJ_DETECTION_MODEL" , DefaultValue : "default_path" , Description : "path to the model file" },
504504 {Name : "CUSTOM_MODEL_PATH" , DefaultValue : "/home/arduino/.arduino-bricks/ei-models" , Description : "path to the custom model directory" },
505505 },
506+ RequireModel : true ,
506507 },
507508 {
508- ID : "arduino:weather_forecast" ,
509- Name : "Weather Forecast" ,
510- Category : "miscellaneous" ,
511- ModelName : "" ,
509+ ID : "arduino:weather_forecast" ,
510+ Name : "Weather Forecast" ,
511+ Category : "miscellaneous" ,
512+ ModelName : "" ,
513+ RequireModel : false ,
512514 },
513515 },
514516 }
@@ -577,6 +579,7 @@ func TestAppBrickInstanceModelsDetails(t *testing.T) {
577579 require .Equal (t , "installed" , res .Status )
578580 require .Empty (t , res .ModelID )
579581 require .Empty (t , res .CompatibleModels )
582+ require .False (t , res .RequireModel )
580583 },
581584 },
582585 {
@@ -597,6 +600,7 @@ func TestAppBrickInstanceModelsDetails(t *testing.T) {
597600 require .Len (t , res .CompatibleModels , 2 )
598601 require .Equal (t , "yolox-object-detection" , res .CompatibleModels [0 ].ID )
599602 require .Equal (t , "face-detection" , res .CompatibleModels [1 ].ID )
603+ require .True (t , res .RequireModel )
600604 },
601605 },
602606 {
@@ -618,6 +622,7 @@ func TestAppBrickInstanceModelsDetails(t *testing.T) {
618622 require .Len (t , res .CompatibleModels , 2 )
619623 require .Equal (t , "yolox-object-detection" , res .CompatibleModels [0 ].ID )
620624 require .Equal (t , "face-detection" , res .CompatibleModels [1 ].ID )
625+ require .True (t , res .RequireModel )
621626 },
622627 },
623628 }
0 commit comments