@@ -414,24 +414,27 @@ def clear_privileges(client)
414
414
def clear_ml_jobs ( client )
415
415
client . ml . close_job ( job_id : '_all' , force : true )
416
416
client . ml . get_jobs [ 'jobs' ] . each do |d |
417
- client . ml . delete_job ( job_id : d [ 'job_id' ] )
417
+ client . ml . delete_job ( job_id : d [ 'job_id' ] , wait_for_completion : false , force : true )
418
418
end
419
419
end
420
420
421
421
def clear_datafeeds ( client )
422
422
client . ml . stop_datafeed ( datafeed_id : '_all' , force : true )
423
423
client . ml . get_datafeeds [ 'datafeeds' ] . each do |d |
424
- client . ml . delete_datafeed ( datafeed_id : d [ 'datafeed_id' ] )
424
+ client . ml . stop_datafeed ( datafeed_id : d [ 'datafeed_id' ] , force : true )
425
+ begin
426
+ client . ml . delete_datafeed ( datafeed_id : d [ 'datafeed_id' ] , force : true )
427
+ rescue StandardError => e
428
+ logger . info ( e )
429
+ end
425
430
end
426
431
end
427
432
428
433
def clear_tasks ( client )
429
434
tasks = client . tasks . get [ 'nodes' ] . values . first [ 'tasks' ] . values . select do |d |
430
435
d [ 'cancellable' ]
431
- end . map do |d |
432
- "#{ d [ 'node' ] } :#{ d [ 'id' ] } "
433
436
end
434
- tasks . each { |t | client . tasks . cancel task_id : t }
437
+ tasks . each { |d | client . tasks . cancel ( task_id : " #{ d [ 'node' ] } : #{ d [ 'id' ] } " ) }
435
438
end
436
439
437
440
def clear_machine_learning_indices ( client )
@@ -448,7 +451,7 @@ def clear_index_templates(client)
448
451
449
452
def clear_transforms ( client )
450
453
client . transform . get_transform ( transform_id : '*' ) [ 'transforms' ] . each do |transform |
451
- client . transform . delete_transform ( transform_id : transform [ 'id' ] )
454
+ client . transform . delete_transform ( transform_id : transform [ 'id' ] , force : true )
452
455
end
453
456
end
454
457
@@ -482,7 +485,7 @@ def delete_trained_models(client)
482
485
return unless models [ 'trained_model_configs' ]
483
486
484
487
models [ 'trained_model_configs' ] . each do |model |
485
- client . ml . delete_trained_model ( model_id : model [ 'model_id' ] , force : true , ignore : 400 )
488
+ client . ml . delete_trained_model ( model_id : model [ 'model_id' ] , force : true , ignore : 400 , timeout : '120s' )
486
489
end
487
490
end
488
491
end
0 commit comments