@@ -414,24 +414,24 @@ 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
- client . ml . stop_datafeed ( datafeed_id : '_all' , force : true )
423
- client . ml . get_datafeeds [ 'datafeeds' ] . each do |d |
424
- client . ml . delete_datafeed ( datafeed_id : d [ 'datafeed_id' ] )
422
+ Thread . new do
423
+ client . ml . stop_datafeed ( datafeed_id : '_all' , force : true , timeout : '120s' )
424
+ client . ml . get_datafeeds [ 'datafeeds' ] . each do |d |
425
+ client . ml . delete_datafeed ( datafeed_id : d [ 'datafeed_id' ] , force : true )
426
+ end
425
427
end
426
428
end
427
429
428
430
def clear_tasks ( client )
429
431
tasks = client . tasks . get [ 'nodes' ] . values . first [ 'tasks' ] . values . select do |d |
430
432
d [ 'cancellable' ]
431
- end . map do |d |
432
- "#{ d [ 'node' ] } :#{ d [ 'id' ] } "
433
433
end
434
- tasks . each { |t | client . tasks . cancel task_id : t }
434
+ tasks . each { |d | client . tasks . cancel ( task_id : " #{ d [ 'node' ] } : #{ d [ 'id' ] } " ) }
435
435
end
436
436
437
437
def clear_machine_learning_indices ( client )
@@ -448,7 +448,7 @@ def clear_index_templates(client)
448
448
449
449
def clear_transforms ( client )
450
450
client . transform . get_transform ( transform_id : '*' ) [ 'transforms' ] . each do |transform |
451
- client . transform . delete_transform ( transform_id : transform [ 'id' ] )
451
+ client . transform . delete_transform ( transform_id : transform [ 'id' ] , force : true )
452
452
end
453
453
end
454
454
@@ -482,7 +482,7 @@ def delete_trained_models(client)
482
482
return unless models [ 'trained_model_configs' ]
483
483
484
484
models [ 'trained_model_configs' ] . each do |model |
485
- client . ml . delete_trained_model ( model_id : model [ 'model_id' ] , force : true , ignore : 400 )
485
+ client . ml . delete_trained_model ( model_id : model [ 'model_id' ] , force : true , ignore : 400 , timeout : '120s' )
486
486
end
487
487
end
488
488
end
0 commit comments