@@ -148,8 +148,7 @@ public async Task AutoMLExperiment_return_current_best_trial_when_ct_is_canceled
148148            res . Metric . Should ( ) . BeGreaterThan ( 0 ) ; 
149149        } 
150150
151-         //https://github.com/dotnet/machinelearning/issues/7203 
152-         [ X86X64Fact ( "Fails on Linux Arm machines" ) ] 
151+         [ Fact ] 
153152        public  async  Task  AutoMLExperiment_finish_training_when_time_is_up_Async ( ) 
154153        { 
155154            var  context  =  new  MLContext ( 1 ) ; 
@@ -167,12 +166,12 @@ public async Task AutoMLExperiment_finish_training_when_time_is_up_Async()
167166                      { 
168167                          var  channel  =  serviceProvider . GetService < IChannel > ( ) ; 
169168                          var  settings  =  serviceProvider . GetService < AutoMLExperiment . AutoMLExperimentSettings > ( ) ; 
170-                           return  new  DummyTrialRunner ( settings ,  0 ,  channel ) ; 
169+                           return  new  DummyTrialRunner ( settings ,  1 ,  channel ) ; 
171170                      } ) 
172171                      . SetTuner < RandomSearchTuner > ( ) ; 
173172
174173            var  cts  =  new  CancellationTokenSource ( ) ; 
175-             cts . CancelAfter ( 10  *  1000 ) ; 
174+             cts . CancelAfter ( 100  *  1000 ) ; 
176175
177176            var  res  =  await  experiment . RunAsync ( cts . Token ) ; 
178177            res . Metric . Should ( ) . BeGreaterThan ( 0 ) ; 
@@ -446,6 +445,7 @@ public void Dispose()
446445        public  async  Task < TrialResult >  RunAsync ( TrialSettings  settings ,  CancellationToken  ct ) 
447446        { 
448447            _logger . Info ( "Update Running Trial" ) ; 
448+             ct . ThrowIfCancellationRequested ( ) ; 
449449            await  Task . Delay ( _finishAfterNSeconds  *  1000 ,  ct ) ; 
450450            ct . ThrowIfCancellationRequested ( ) ; 
451451            _logger . Info ( "Update Completed Trial" ) ; 
0 commit comments