diff --git a/ax/storage/sqa_store/utils.py b/ax/storage/sqa_store/utils.py index 3fc233ff15e..0fab599e094 100644 --- a/ax/storage/sqa_store/utils.py +++ b/ax/storage/sqa_store/utils.py @@ -27,6 +27,7 @@ "_best_arm_predictions", "_adapter_kwargs", "_candidate_metadata_by_arm_signature", + "_cached_session", "_curr", "_experiment", "_gen_metadata",