Thanks for your response. Do you know why it is returning a None object? I used the same code roughly 6 months ago and it worked perfectly?
I thought it may be an issue that the state_dicts are not saving in the file path and so a there is nothing to ‘get’ when calling get_best_trial.