-ji Z%SrSSKJr SSKrSSKJr SSKJr SSKrSSKJr SSK r SSK r SSK r SSK r SSK r SSKJr SSKrSSKrSSKrSSKJr SS KJr SS KJr SS KJr SS KJr SS KJr SSKJr SSKJr SSKJ r SSK!J"r" SSK!J#r# SSK$J%r% \"S5r&Sr'SDSjr(SESjr)SFSjr*SGSjr+"SS\5r,"SS5r-SHSjr.SHSjr/SIS jr0"S!S"5r1"S#S$\15r2"S%S&\15r3"S'S(\15r4"S)S*\15r5"S+S,\15r6"S-S.\15r7"S/S0\15r8"S1S2\15r9"S3S4\15r:"S5S6\15r;"S7S8\15r<\2\3\4\5\6\7\8\9\:\;\S;'SJS<jr?SKS=jr@SLS>jrASMSNS?jjrBSOS@jrCSPSAjrDSPSBjrESQSCjrFg)RzfOptuna CLI module. If you want to add a new command, you also need to update the constant `_COMMANDS` ) annotationsN)ArgumentParser) Namespace)Enum)Any) _LazyImport) optuna_warn) CLIUsageError)ExperimentalWarning) BaseStorage)JournalFileStorage)JournalRedisStorage)JournalStorage) RDBStorage)JournalFileBackend)JournalRedisBackend) TrialStatezoptuna.study._dataframez%Y-%m-%d %H:%M:%ScUbU$[RRS5nUb[S[5 U$[ S5e)NOPTUNA_STORAGEzSpecifying the storage url via 'OPTUNA_STORAGE' environment variable is an experimental feature. The interface can change in the future.zStorage URL is not specified.)osenvirongetr r r ) storage_url env_storages D/home/james-whalen/.local/lib/python3.13/site-packages/optuna/cli.py_check_storage_urlr*sL**..!12K S   7 88c[U5nU(aU[R:Xa[[U55$U[R:Xa[[ U55$U[ R:Xa[[ U55$U[ R:Xa[[ U55$U[R:Xa [U5$[S5eURS5(a[[U55$[RRU5(a[[ U55$[U5$![RRa [S5ef=f)NzUnsupported storage classredisz.Failed to guess storage class from storage_url)rr__name__rrrr rr startswithrpathisfile sqlalchemyexc ArgumentError)r storage_classs r _get_storager(9s2$[1K /88 8!"5k"BC C /88 8!"5k"BC C .77 7!"4["AB B .77 7!"4["AB B J// /k* *788g&&1+>?? ww~~k""0=>>N+&& >> ' 'NLMMNs E *E5cUb[U[[45(aU$[U[R5(aUR [ 5$[U[ 5(a[ SU55$[U[5(a[SU55$[U[5(a8UR5VVs0sHup[U5[U5_M snn$[U5$s snnf)Nc38# UHn[U5v M g7fN _format_value.0vs r _format_value..Ys4eM!$$ec38# UHn[U5v M g7fr+r,r.s rr1r2[s5u!]1%%ur3) isinstanceintfloatdatetimestrftime_DATETIME_FORMATlisttupledictitemsr-str)valuekr0s rr-r-Rs } 53,77 E8,, - -~~.// E4 4e444 E5 ! !5u555 E4 ?D{{}M}tq a -"22}MM5zNs !C;c^ /n/nU(GaRUHm T SS:waURT SST S35 M*[U 4SjU55(aXSnUH#nT U;dM [U[UT 55nM% [ U5HnURT SSU35 M MURT S5 M UHn0nUHwm T U;aM [ UT 5n T SS:waXT SST S3'M3[U 4SjU55(a"[ U 5HupzXT SSU3'M MoXT S'My URU5 M XC4$UH"m T SU;dMURT S5 M$ UHnUV s0sHo0_M n n UHm T U;aM [ UT 5n [T S[5(aYU T S0:Xa/U T S'U T S==S/[T SS-[U T S5- S5-- ss'XT ST S'MT SS:waXT ST S'MXT S'M URU 5 M XC4$s sn f)Nr_c3n># UH*n[URT5[[45v M, g7fr+r5rr;r<r/recordcolumns rr1#_convert_to_dict..ks*YQXvZ 6 2T5MBBQX25c3n># UH*n[URT5[[45v M, g7fr+rGrHs rr1rK|s*]U\6FJJv$6u FFU\rL) appendanymaxlenranger- enumerater5r6)recordscolumnsflattenheaderret max_lengthrIirowr@r0 column_nameattrsrJs @r_convert_to_dictr^bsF CFayB  1VAYK89YQXYYY %F'%(S5H%I &z*AMMVAYKq"45+ fQi(FC!'%fVn5!9?6;6!9+Qvayk23]U\]]] )% 023vayk1#./!1+06!9+'" JJsOJ ;-Fay& fQi(FHN$O"_E$O!'%fVn5fQi--VAY'2-+-fQi(&)$VAY]SvVWyIYEZ5Z\]1^(^^$27&)$VAY/AY"_27&)$VAY/',&)$" JJu #& ;%%Ps I&c \rSrSrSrSrSrSrg) ValueTyperrCN)r __module__ __qualname____firstlineno__NONENUMERICSTRING__static_attributes__rcrrr`r`s DG Frr`c<\rSrSrSSjrS SjrS SjrS SjrSrg) CellValuecXlUc[RUlg[ U[ [ 45(a[RUlg[RUlgr+) r@r`rg value_typer5r6r7rhri)selfr@s r__init__CellValue.__init__sB ='nnDO U| , ,'//DO'..DOrc[UR[R5(aURR[5$[ UR5$r+)r5r@r8r9r:r?rps r__str__CellValue.__str__s= djj("3"3 4 4::&&'78 8tzz? "rc>[[UR55$r+)rQr?r@rts rwidthCellValue.widths3tzz?##rc[UR5nURcSU-$U[R:XaUSU3$USU3$)N ><)r?r@r`rh)rprorxr@s r get_stringCellValue.get_stringsSDJJ :: ;  9,, ,AeW9% &AeW9% &r)r@roN)r@rreturnNone)rr?rr6)ror`rxr6rr?) r rdrerfrqrurxr~rjrcrrrlrls/# $'rrlc/nUH^n/nUH3nURURU5b [X55OS5 M5 URSRU55 M` SRU5$)NrDr{ )rNrr?join)rTrWvaluesrIr[r\s r _dump_valuersn F!K JJ6::k3J3Vs6./\^ _ "  chhsm$ 99V rc ^/nUHHn/nUH,nUR[URU555 M. URU5 MJ SnSnUVs/sHnSPM n n[[ U55GH%mUVs/sHoDTR PM n n[ Rn U H)n U [ R:XdM[ Rn M+ [ U5S:Xa[ UT5n O+[[ UT5[U4SjU555n USU S--S-- nU [ R:XaUSUTSU 3S 3- nOUSUTS U 3S 3- n[U5H)upX==SUTRX5-S -- ss'M+ GM( S nXS -- nXS -- nXS -- nU H nUUS -- nM XS -- nU$s snfs snf) N+|rc3H># UHoTR5v M g7fr+)rx)r/r[rJs rr1_dump_table..s!4YTXS[5F5F5H5HTXs"-rbr{r|z |r}rDr) rNrlrrRrQror`rhrirPrSr~)rTrWrowsrIr[r\ separator header_stringrE rows_string value_typesrot max_widthrZrX row_stringrJs @r _dump_tablers D!K JJyK!89 :" C  IM $%13K%F $9=>#6{-- >&& AI$$$&--  t9>F6N+ICv/4YTX4Y1YZISIM*S00 ** * q)} =R@ @M q)} =R@ @MoFA NcCK$:$::$QQTXX XN&%$ Ct C4 Ct C!  zD  "t C J7&>s  GGc[U[5(a[XU5upEO[U/X5upEUS:Xa[XE5R 5$US:Xa[ XE5R 5$US:Xa`[U[5(a$[ R"U5R 5$[ R"US5R 5$US:Xa`[U[5(a$[R"U5R 5$[R"US5R 5$[SUS35e)Nr@tablejsonryamlz"Optuna CLI does not supported the z format.) r5r;r^rstriprrdumpsr safe_dumpr )rTrU output_formatrVrrWs r_format_outputrs '4  )'GD)7)WF6*0022 ' !6*0022 & gt $ $::f%++- -::fQi(..0 0 & gt $ $>>&)//1 1>>&),224 4@xXYYrc6\rSrSrSrSSjrS SjrS SjrSrg) _BaseCommandizBase class for commands. Note that command classes are not intended to be called by library users. They are exclusively used within this file to manage Optuna CLI commands. cT[RR[5Ulgr+)optunalogging get_loggerr loggerrts rrq_BaseCommand.__init__snn//9 rcg)zmAdd arguments required for each command. Args: parser: `ArgumentParser` object to add arguments Nrcrpparsers r add_arguments_BaseCommand.add_argumentss rc[e)zDefine action if the command is called. Args: parsed_args: `Namespace` object including arguments specified by user. Returns: Running status of the action. 0 if this method finishes normally, otherwise 1. )NotImplementedError)rp parsed_argss r take_action_BaseCommand.take_action$s "!r)rN)rrrrrrrrrr6) r rdrerf__doc__rqrrrjrcrrrrs :  "rrc,\rSrSrSrSSjrSSjrSrg) _CreateStudyi3zCreate a new study.c URSSSS9 URSS[SSS9 URSS S S S 9 URS [SSSSS9 g)N --study-namez?A human-readable name of a study to distinguish it from others.defaulthelpz --direction)minimizemaximizeznSet direction of optimization to a new study. Set 'minimize' for minimization and 'maximize' for maximization.)rtypechoicesrz--skip-if-existsF store_truezgIf specified, the creation of the study is skipped without any error when the study name is duplicated.ractionrz --directionszSet directions of optimization to a new study. Put whitespace between directions. Each direction should be either "minimize" or "maximize".r)rrrrnargs add_argumentr?rs rr_CreateStudy.add_arguments6s R   ,@    C    ,0  rc[URUR5n[R"UUR UR URURS9R n[U5 g)N)storage study_name direction directionsload_if_existsr) r(rr'r create_studyrrrskip_if_existsprint)rprrrs rr_CreateStudy.take_actionVsh{22K4M4MN(("--!++"--&55  *  jrrcNrrr rdrerfrrrrjrcrrrr3s @ rrc,\rSrSrSrSSjrSSjrSrg) _DeleteStudyiczDelete a specified study.c&URSSSS9 g)Nrz The name of the study to delete.rrrs rr_DeleteStudy.add_argumentsfsND?abrc[URUR5nURUR5nUR U5 g)Nr)r(rr'get_study_id_from_namer delete_study)rprrstudy_ids rr_DeleteStudy.take_actionisA{22K4M4MN11+2H2HIX&rrcNrrrrcrrrrcs#crrc,\rSrSrSrSSjrSSjrSrg) _StudySetUserAttributeipz Set a user attribute to a study.clURSSSS9 URSSSSS9 URSSS S9 g) NrTz3The name of the study to set the user attribute to.)requiredrz--keyz-kzKey of the user attribute.z--valuezValue to be set.rrs rr$_StudySetUserAttribute.add_argumentsssN F  GTD?[\I;MNrc[URUR5n[R"X!R S9nUR URUR5 URRS5 g)NrrzAttribute successfully written.r) r(rr'r load_studyr set_user_attrkeyr@rinfo)rprrstudys rr"_StudySetUserAttribute.take_action|s`{22K4M4MN!!'>T>TU KOO[->->? :;rrcNrrrrcrrrrps*Orrc,\rSrSrSrSSjrSSjrSrg) _StudyNamesiz1Get all study names stored in a specified storagec 4URSS[SSSS9 g)N-f--formatr@rrrr@Output format.rrrrrrs rr_StudyNames.add_argumentss'  6!  rc [URUR5n[R"U5n/nSnUHnUR XV05 M [ [XE/URSS95 g)NnamerDF)rVr) r(rr'rget_all_study_namesrNrrformat)rprrall_study_namesrT record_keyrs rr_StudyNames.take_actionsj{22K4M4MN 44W=! )J NNJ3 4* nWlK4F4FPUVWrrcNrrrrcrrrrs; rrc4\rSrSrSr/SQrSSjrS SjrSrg) _StudiesizShow a list of studies.)rrrDn_trialsrDdatetime_startrDc XURSS[SSSS9 URSSS S S 9 g) Nrrrrrr --flattenFrz*Flatten nested columns such as directions.rrrs rr_Studies.add_argumentssH  6!    =  rcr[URUR5n[R"USS9n/nUHnUR bUR R [5OSn0nURUS'[SUR55US'URUS'XgS'URUS'URU5 M [S U55(aURRS5 [![#X@RUR$UR&55 g ) NF)include_best_trialrc38# UHoRv M g7fr+)r)r/ds rr1'_Studies.take_action..s-Klfflr3rrr user_attrsrDc30# UH oS0:gv M g7f)r Nrc)r/rs rr1rs 00+2D2DkFYFY  rrcNrr) r rdrerfrrrrrjrcrrrrs! rrc,\rSrSrSrSSjrSSjrSrg) _TrialsizShow a list of trials.c URS[SSS9 URSS[SSS S 9 URS S S SS9 g)NrTz,The name of the study which includes trials.rrrrrrrrrrFr5Flatten nested columns such as params and user_attrs.rrrs rr_Trials.add_argumentsi ?    6!    H  rcd[S[5 [URUR5n[ R "X!RS9nSUR5(dSOSSSSS S S 4n[RX45upV[[XVURUR55 g ) NzP'trials' is an experimental CLI command. The interface can change in the future.rnumberr@rrdatetime_completedurationparamsr stater)r r r(rr'rrr_is_multi_objective _dataframe$_create_records_and_aggregate_columnrrrrVrprrrr]rTrUs rr_Trials.take_actions ^  {22K4M4MN!!'>T>TU  4466GH        &JJ5X nW{/A/A;CVCVWXrrcNrrrrcrrrrs  ,rrc,\rSrSrSrSSjrSSjrSrg) _BestTrializShow the best trial.c URS[SSS9 URSS[SSS S 9 URS S S SS9 g)NrTz,The name of the study to get the best trial.rrrrrrrrFrrrrrs rr_BestTrial.add_arguments rrc[S[5 [URUR5n[ R "X!RS9nSUR5(dSOSSSSS S S 4n[RX45upV[[XSRRXaRUR 55 g ) NzT'best-trial' is an experimental CLI command. The interface can change in the future.rrr@rrrrrr rr)r r r(rr'rrrr r!r"rr best_trialrrrVr#s rr_BestTrial.take_action!s b  {22K4M4MN!!'>T>TU  4466GH        &JJ5X ((//0';M;M{ObOb  rrcNrrrrcrrr&r&s ,rr&c,\rSrSrSrSSjrSSjrSrg) _BestTrialsi=z2Show a list of trials located at the Pareto front.c URS[SSS9 URSS[SSS S 9 URS S S SS9 g)NrTzJThe name of the study to get the best trials (trials at the Pareto front).rrrrrrrrFrrrrrs rr_BestTrials.add_arguments@si ]    6!    H  rc^ [S[5 [URUR5n[ R "X!RS9nURVs/sHoDRPM snm SUR5(dSOSSSSS S S 4n[RX55upg[[U 4S jU55n[[!XUR"UR$55 g s snf)NzU'best-trials' is an experimental CLI command. The interface can change in the future.rrr@rrrrrr rc>UST;$)NrrDrc)rI best_trialss r)_BestTrials.take_action..ls&2HK2Wrr)r r r(rr'rrrr3rr r!r"r;filterrrrrV) rprrrtrialr]rTrU best_recordsr3s @rr_BestTrials.take_actionVs   {22K4M4MN!!'>T>TU161B1BC1B||1BC  4466GH        &JJ5XF#WY`ab  n\K4F4F H[H[\]DsC2rcNrrrrcrrr-r-=s< ,rr-c"\rSrSrSrSSjrSrg)_StorageUpgradeiqz%Upgrade the schema of an RDB storage.c[UR5n[USSS9nUR5nUR5nUR5nXE:XaUR RS5 g XF;aGUR RS5 UR5 UR RS5 g [S5 g ![RR a UR RS5 gf=f) NT)skip_compatibility_checkskip_table_creationzInvalid RDBStorage URL.rCzThis storage is up-to-date.z3Upgrading the storage schema to the latest version.z!Completed to upgrade the storage.zYour optuna version seems outdated against the storage version. Please try updating optuna to the latest version by `$ pip install -U optuna`.r)rrrr$r%r&rerrorget_current_versionget_head_versionget_all_versionsrupgrader )rprrrcurrent_version head_versionknown_versionss rr_StorageUpgrade.take_actionts()<)<=   dPTG "557//1  113  * KK  : ; . KK  R S OO  KK  @ A -  %~~++  KK  7 8 s C 9DDrcNr)r rdrerfrrrjrcrrr;r;qs /rr;c,\rSrSrSrSSjrSSjrSrg) _Askiz*Create a new trial and suggest parameters.c URS[SS9 URS[SS9 URS[SS9 URS[S S9 URS S [S S SS9 URSSSSS9 g)NrName of study.rrz --samplerz'Class name of sampler object to create.z--sampler-kwargsz8Sampler object initialization keyword arguments as JSON.z--search-spaceztSearch space as JSON. Keys are names and values are outputs from :func:`~optuna.distributions.distribution_to_json`.rrrrrrrFrz&Flatten nested columns such as params.rrrs rr_Ask.add_argumentssN;KLKc8ab K   F     6!    9  rc ([S[5 [URUR5nUUR SS.nUR baURb![R"UR5nO0n[[RUR 5nU"S0UD6nXcS'OURb [S5eURbu[R"UR5R5VVs0sH9upxU[R R#[R$"U55_M; n nnO0n [R&"USUSUR)S5S9n U R-U S 9n UR.R1S U R2S U R4S 35 SU R20n S/n [7U R45S:Xa(UR8(d0U S'U R;S5 O distributionsjson_to_distributionrrrKeyErroraskrrrrrQrVrNrrr)rprrcreate_study_kwargsrR sampler_clsrOrdistrWrr7rIrU param_name param_values rr_Ask.take_actionsk [  {22K4M4MN%00"     *))5!%K,F,F!G!#!&//;3F3FGK!3N3G-4 *))5 8  # # / #'**[-E-E"F"L"L"N"NJDf**?? 4@PQQ"N L L %%.|<+I6+// :E l ; < ~5Fu||nTUVW. " NN> *+0<<+=+=+?' 1<*-.*56,@ nVk.@.@+BUBUVWG D  sAI5 +I;;JrcNrrrrcrrrIrIs4 >DrrIc,\rSrSrSrSSjrSSjrSrg) _Telliz5Finish a trial, which was created by the ask command.cURS[SS9 URS[SS9 URS[SSS 9 URS [S S S 9 URSSSSS9 g)NrrKrLz--trial-numberz Trial number.z--valuesrzObjective values.)rrrz--statez Trial state.)completeprunedfail)rrrz--skip-if-finishedFrzSIf specified, tell is skipped without any error when the trial is already finished.r)rr?r6r7rs rr_Tell.add_argumentssN;KL,3_MJU#DWX 2     rc [S[5 [URUR5n[ R "UURS9nURb"[URR5nOSnURnURnURUUUURS9 URR!SUSUSUS35 g) NzN'tell' is an experimental CLI command. The interface can change in the future.r)r7rrskip_if_finishedz Told trial z with values z and state rQr)r r r(rr'rrrrrupper trial_numberrtellrjrr)rprrrrrlrs rr_Tell.take_action s \  {22K4M4MN!!"--     ('1+2C2C2I2I2K'LEE"// ## (99   ;|nM&UZT[[\]^rrcNrrrrcrrrcrcs? $rrc) z create-studyz delete-studyzstudy set-user-attrz study-namesstudiestrialsz best-trialz best-trialszstorage upgrader[rmzdict[str, type[_BaseCommand]] _COMMANDSc[R[R[R/n[R[ R/nXU-;aU$[ R"SUS[U5SSS35e)NzInvalid choice: z (choose from rC)) rr rrr rargparseArgumentTypeErrorr?)r@rdeprecated_choicess r:_parse_storage_class_without_suggesting_deprecated_choicesrx=s##$$G ##$$ ,,,  $ $ 5'Wa1C0DAF rc  URSSSS9 URSSS[S9 UR5nURSSS S S S S 9 URSSSS SSS9 URSSSSS9 URSSSSS9 U$)Nz --storagezbDB URL. (e.g. sqlite:///example.db) Also can be specified via OPTUNA_STORAGE environment variable.rz--storage-classz,Storage class hint (e.g. JournalFileBackend))rrrz-vz --verbosecount verbose_levelrCz.Increase verbosity of output. Can be repeated.)rdestrrz-qz--quiet store_constrz+Suppress output except warnings and errors.)rr|constrz --log-filestorez2Specify a file to log output. Disabled by default.)rrrz--debugFrzShow tracebacks on errors.r)rrxadd_mutually_exclusive_group)r verbose_groups r_add_common_argumentsrNs  M   ; G  779M   =    :   A    )  Mrcd^TR5n0n[R5H_upEU"5nURXA/[R "U5S9nUR U5 URURS9 XsU'Ma SU4SjjnURSSS9RUS9 U$)N)parentsr)handlerc&>TR5 gr+) print_help)args main_parsers r _print_help"_add_commands.._print_helps rrzShow help message and exit.)rrrrr) add_subparsersrqr> add_parserinspectgetdocr set_defaultsr) r parent_parser subparserscommand_name_to_subparser command_name command_typecommand subparserrs ` r _add_commandsr}s++-J "&/oo&7" .)) /|8T*  i(w':':;2;,/'8!&'DERRS %$rc[SS9n[U5n[X/S9nURSSSRS[R 5S9 [ X!5nX#4$) NF)add_help) descriptionrz --versionversionz{0} {1}r)rr)rrrrr __version__r)rrrrs r _get_parserrsg#E2M)-8M [/RKIy/?/?&J\J\/]!.k I  11rc[U5S:aUSSOS/n[[U55HOn[XS-5H:nSRXUS-5nU[;dM%USUXS-S-nU/U-s s $ MQ U$)NrCrrbr{)rQrRrrq)argvrZjcommand_candidateoptionss r_preprocess_argvrs4y1}486(D 3t9 qa%A #!a% 9  I-r(Ta%']2)*W44 ! Krc@[R"5nUR[R5 [R"[ R 5n[R[R[RS.RUR[R5nURU5 UR[RR55 URU5 [RRU5 g)N)rrCrb)r getLoggersetLevelDEBUG StreamHandlersysstderrWARNINGINFOrr{ setFormatterrcreate_default_formatter addHandler set_verbosity)r root_loggerstream_handler logging_levels r_set_verbosityrs##%K'**3::6N ?? << == c$  gmm,  M* G G IJ>* NN  /rc@URcg[R"5nUR[R5 [R "URS9nUR [RR55 URU5 g)N)filename) log_filerrrr FileHandlerrrrr)rr file_handlers r _set_log_filersq }}##%K'&&LfnnEEGH<(rc6[5up[Rn[U5nUR U5n[ U5 [ U5 [R"S5nURU5$![aSnUR(aURU5 O&URU5 XSR5 SnAgSnAf[a< SR!USS5nURSUS35 UR5 gf=f)NrrrCr{'z/' is not an optuna command. see 'optuna --help')rrrr parse_argsrrrrrr debug exceptionr?rAttributeErrorr)rrrpreprocessed_argvrreargv_strs rmainrs-8]*K 88D(.  ! !"3 4D4$   x (F||D!!  ::   Q  LLO &&: ; F F H 88DH% q "QRS  s &A77 DA CADD)r str | Nonerr?)rrr'rrr )r@rrr)rTz list[dict[tuple[str, str], Any]]rUlist[tuple[str, str]]rVboolrz&tuple[list[dict[str, Any]], list[str]])rTzlist[dict[str, Any]]rW list[str]rr?) rTz=list[dict[tuple[str, str], Any]] | dict[tuple[str, str], Any]rUrrr?rVrrr?)r@r?rr?)rrrr)rrrrrzdict[str, ArgumentParser])rD)rr?rz0tuple[ArgumentParser, dict[str, ArgumentParser]])rrrrrr)Gr __future__rrurrr8enumrrrrrrtypingrsqlalchemy.excr$rroptuna._importsroptuna._warningsr optuna.exceptionsr r optuna.storagesr r rrroptuna.storages.journalrr optuna.trialrr!r:rr(r-r^r`rlrrrrrrrrrrr&r-r;rIrcrq__annotations__rxrrrrrrrrcrrrs ##   '(+1'./*&67#2 3 & 9N2  7 -78M7X\7+7t '': %PZ JZ "ZZ Z  Z:""D-<-` < \.,04|4n/l/d22j1,1hl%%0 2  0$ )r