def create_new_study(self, study_name: Optional[str] = None) -> int:
session = self.scoped_session()
if study_name is None:
study_name = self._create_unique_study_name(session)
study = models.StudyModel(study_name=study_name, direction=StudyDirection.NOT_SET)
session.add(study)
if not self._commit_with_integrity_check(session):
raise optuna.exceptions.DuplicatedStudyError(
"Another study with name '{}' already exists. "
"Please specify a different name, or reuse the existing one "
"by setting `load_if_exists` (for Python API) or "
"`--skip-if-exists` flag (for CLI).".format(study_name)
)
_logger.info("A new study created in RDB with name: {}".format(study.study_name))
return study.study_id
def create_new_study(self, study_name: Optional[str] = None) -> int:
try:
with _create_scoped_session(self.scoped_session) as session:
if study_name is None:
study_name = self._create_unique_study_name(session)
direction = models.StudyDirectionModel(
direction=StudyDirection.NOT_SET, objective=0
)
study = models.StudyModel(study_name=study_name, directions=[direction])
session.add(study)
except IntegrityError:
raise optuna.exceptions.DuplicatedStudyError(
"Another study with name '{}' already exists. "
"Please specify a different name, or reuse the existing one "
"by setting `load_if_exists` (for Python API) or "
"`--skip-if-exists` flag (for CLI).".format(study_name)
)
_logger.info("A new study created in RDB with name: {}".format(study_name))
return self.get_study_id_from_name(study_name)