i\ SrSSKrSSKrSSKrSSKJr SSKJrJrJ r J r J r SSK J r Jr SSKJrJrJr SSKJr SSKJr SS KJrJrJr \R6"S 5"S S 55r\R6"S 5"S S55rS\S\ \S\ \S\SS4 SjrS\S\ \S\ \S\SS4 Sjr S\S\ \S\ \S\SS4 Sjr!\R6"S 5"SS55r"g)z) Common test functions testing launchers N)Path)AnyCallableListOptionalSet) DictConfig OmegaConf)markparamraises) TaskFunction)HydraException) TSweepRunnerintegration_testverify_dir_outputshydra_restore_singletonsc x\rSrSrS\\\/\44SjrS\S\ S\ \ S\ SS4 S jr S\S\ S\ \ S\ SS4 S jr S\S\ S\ \ S\ SS4 S jrS\S\ S\ \ S\ SS4 S jrS\S\ S\ \ S\ SS4 S jrS\S\ S\ \ S\ SS4 SjrS\S\ S\ \ S\ SS4 SjrS\S\ S\ \ S\ SS4 SjrS\S\ S\ \ S\ SS4 SjrSrg)LauncherTestSuitereturnc(S[S[4SjnU$)N_rcg)Nd)rs e/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/hydra/test_utils/launcher_common_tests.py task_func6LauncherTestSuite.get_task_function..task_funcs)r r)selfrs rget_task_function#LauncherTestSuite.get_task_functions   r hydra_sweep_runner launcher_name overridestmpdirNcF[USU-/U-UR5US9 gNhydra/launcher=r& task_functiontemp_dir sweep_1_jobr"r!r$r%r&r's rtest_sweep_1_job"LauncherTestSuite.test_sweep_1_job.  (=89IE002  r cF[USU-/U-UR5US9 gr)) sweep_2_jobsr"r0s rtest_sweep_2_jobs#LauncherTestSuite.test_sweep_2_jobs-s.  (=89IE002  r c [[[R"S5S9 U"SSSSSUSU-SS/-US 9 SSS5 SSS5 g!,(df  N=f!,(df  g=f) NzQSweeping over Hydra's configuration is not supported : 'hydra.verbose=true,false')matchhydra.test_utils.a_moduleconfigs compose.yamlr*+a=0,1zhydra.verbose=true,false calling_filecalling_moduler, config_path config_namer&r-)r rreescaper0s r!test_not_sweeping_hydra_overrides3LauncherTestSuite.test_not_sweeping_hydra_overrides;s ))c $!:"%*#% 5.           s!AA A A A A+cF[USU-/U-UR5US9 gr)r.r0s rtest_sweep_1_job_strict)LauncherTestSuite.test_sweep_1_job_strictXr3r c[[5 [USU-S/U-UR5US9 SSS5 g!,(df  g=f)Nr*zboo=barr+)r Exceptionr/r"r0s r#test_sweep_1_job_strict_and_bad_key5LauncherTestSuite.test_sweep_1_job_strict_and_bad_keyfsDI  ",} group1=file1 group1=file2foo r) r"r createreturnslenranger&cfgr r!r$r%r&r'base_overridessweepexpected_overrides expected_confijob_rets r-test_sweep_over_unspecified_mandatory_default?LauncherTestSuite.test_sweep_over_unspecified_mandatory_defaults,m;=QR"6002!7$0  ..0@A"))5"+6 8H8H%QS8UV ==, ,,u}}Q'(A- --1X--*1-((,>,AAAA{{m&6666"7,=,=> UUs BC44 Dc SU-SS/nU"SSSSSXS-US9nS S/S S//nS S S SS.0S SSSSS.0/nU URce[URS5S:Xde[S5HRn URSU n U RXy:XdeU RX:Xde[ XR5 MT SSS5 g!,(df  g=f)a> Tests that we can override things in the configs merged in only during the sweep config construction db.user=someone does not exist db_conf.yaml, and is only appear when we merge in db=mysql or db=postgresql. This presents a tricky situation when operating in strict mode, this tests verifies it's handled correctly. r*zdb=mysql,postgresqlzdb.user=someoneNr:r;z db_conf.yamlr>zdb=mysqlz db=postgresqldbmysqlsecretsomeone)driverpassworduser postgresqldrowssaprX)rprnrotimeoutrrZ)r\r]r^r&r_rr`s rtest_sweep_and_override)LauncherTestSuite.test_sweep_and_overrides!  - !   #6!&$0 * + / 0  g8YO P%* *!    ==, ,,u}}Q'(A- --1X--*1-((,>,AAAA{{m&6666"7,=,=> UUs BC  Cc VSU-/U-nS[S[4SjnU"SSUSSUUS9nS[4S jn[R"S U5 U URceURS n [ U 5S :XdeU S R S :XdeSSS5 g!,(df  g=f)Nr*crcUR$)N)x)rws rrDLauncherTestSuite.test_sweep_with_custom_resolver..task_funcs 33Jr r:r;custom_resolverr>cg)NrWrrr rmy_custom_resolverMLauncherTestSuite.test_sweep_with_custom_resolver..my_custom_resolversr r}rrW)r rr register_new_resolverr\r] return_value) r!r$r%r&r' overrides1rrbr}rfs rtest_sweep_with_custom_resolver1LauncherTestSuite.test_sweep_with_custom_resolvers(-789D    #6#!)   C  ''(<>PQ ==, ,,mmA&Gw<1$ $$1:**e3 33 UUs AB B(r)__name__ __module__ __qualname____firstlineno__rrrr"rstrrrr1r6rErHrLrPrgrtr__static_attributes__rr rrrs 8HcUCZ,@#A  (    9        (    9      (9     :  (    9      (9     "  (    9      ?(??9 ?  ?  ?80?(0?0?9 0?  0?  0?d4(449 4  4  4r rc 8\rSrSrS\S\S\\S\SS4 SjrS r g) BatchedSweeperTestSuiter$r%r&r'rNc SS/nSU-/nURU5 URU5 U"SSSSSUUS9nSS /SS /SS /S S /S S /S S //nS SS.S SS.S SS.SSS.SSS.SSS./n [5n U URn U ce[U 5S- n U R 5(de[ R "U 5n U RRRU:XdeURce[UR5S:XdeURHn[U5S:XaMe URVVs/sH oHoPM M nnn[U5S:Xde[U5HaunnURUU:XdeURU U:XdeU RUR5 [!UUR5 Mc SSS5 [U 5S:Xdegs snnf!,(df  N&=f)NrSzbar=100,200,300r*r:r;r<r>rUzbar=100zbar=200zbar=300rVrXrrWbari,rY multirun.yamlrZ)extendsetr-rexistsr loadhydrar&taskr\r] enumerater_add working_dirr)r!r$r%r&r' job_overrideshydra_overridesrbrcrddirsr-multirun_cfg_path multirun_cfgbatchrtflatidxrfs rtest_sweep_2_jobs_2_batches3BatchedSweeperTestSuite.test_sweep_2_jobs_2_batchess?./@A ,}<=')"6!& Y ' Y ' Y ' Y ' Y ' Y '  s #s #s #s #s #s #   ~~H' '' $X @ $++-- --$>>*; !> )$ W((,>s,CCCC{{mC&8888,,-"7G,=,=> !0!*4yA~~CUs&3B9G;0G;G5BG;5G;; H r) rrrrrrrrrrrr rrrs;<(<<9 <  <  <r rr$r&r,r-rc U"SSUSSUUS9nU URceURSn[U5S:XdeUSR/:XdeUSRSS S .:XdeUSRR R RS :Xd5S USRR R R-5eUSRR R RceUSRR R Rce[URSS5 SSS5 g!,(df  g=f) z Runs a sweep with one job Nr:r;r<r>rrrXrra_moduleUnexpected job name: ) r\r]r&r_ hydra_cfgrjobnameidnumr)r$r&r,r-rbrfs rr/r/5sR 2#" E }}(((--"7|q   qz##r)))qz~~C!8888qz##))--22j@ #gaj&:&:&@&@&D&D&I&I I @qz##))--00<<<qz##))--11===5==+A./ s D>E E$c "S/nURU5 U"SSUSSUUS9nSSS S .nU URceURce[UR5nUS - nUR 5(de[ R "U5nURRRU:Xde[URS 5S :Xde[S 5GHn URS U n [ R"U5n Xl U RS U 3/:XdeU RU :XdeU RRR R"S:Xd2SU RRR R"-5eU RRR R$ceU RRR R&ce[)XR5 U[+U 5- n UR-5V s/sHoR/5(dMU PM nn U R 5(aGMSU SU35e SSS5 gs sn f!,(df  g=f)z Runs a sweep with two jobs r=Nr:r;composer>rXrr)rWrarrZz+a=rr'z' does not exist, dirs: )rr-r\rrr rrr&rr]r^r[rr_rrrrrrriterdiris_dir)r$r&r,r-rrbbase_cfgrrrerfrdpathrylsts rr5r5VsVJM ]# 2# E#A.H ~~)))}}((('$6 '')))) ~~&78 !!++00MAAA5==#$)))qAmmA&q)G%,,X6MO$$3qc 3 33;;-/ //$$**..33zA ''*;*;*A*A*E*E*J*JJ A$$**..11= ==$$**..22> >> w(9(9 :c!f$D&..0?0HHJ10C?;;== IAdV+CC5"I I= 4@5 s*G>J'I;I; J#J;J Jc 0[R"U5nURS/5 U"SSUSSUUS9nS/S//n[R"S S S .5[R"S S S .5/nU UR ce[ UR S 5S:Xde[S5HRnUR S UnURXW:XdeURXg:Xde[XR5 MT SSS5 g!,(df  g=f)zB Make sure that optimizers=adam,nesterov is interpreted correctly rSNr:r;rr>rUrVrXrrrYrrZ) copydeepcopyrr r[r\r]r^r&r_r) r$r&r,r-rbrcrdrerfs rrOrOs i(I *+, 2# E**^,<=C01C01M }}(((5==#$)))qAmmA&q)G$$(:(== ==;;-"22 22 w(9(9 :  s /BD Dcr\rSrSrS\\4SjrS\S\4SjrS\S\ \ 44Sjr \ R"S\"S /S S S S 9\"S S /S SSS 9\"SSSS000/SSSS 9\"SSSS000S /SSSS 9/5S\S\S\ \ S\ S\ S\S\ \ SS 4Sj5r\ R"S\"SSS S!S".00/S#S$S 9\"0S%S&/S'S(S 9\"SSS S!S".00S%S&/S'S)S 9\"SS*S+S".0S,S-S..S/S0/S1S2S 9\"S*S+S".S3S4S5S6S7/S8.00S9.S,S-S:S;./SS 9/5S\S\ S\ \ S?\ S\S\ \ SS 4S@j5rS\S\S\ \ SS 4SAjrS\S\S\ \ SS 4SBjrSCrg )DIntegrationTestSuiteircg)z By default test applications working dir is tmpdir, override this method if that's not the case. This could be helpful when the tests kick off applications on remote machines. Nr)r!s rget_test_app_working_dir-IntegrationTestSuite.get_test_app_working_dirs r r'cU$)zf By default test applications will use tmpdir provided by the This can be customized by applications. r)r!r's rget_test_scratch_dir)IntegrationTestSuite.get_test_scratch_dirs  r .cPS[[S[S[[4SjnU$)zy By default this does nothing, but it allows custom execution commands. Useful if the tests are not kicked off by python cmdfilenamercU$)z param cmd: old python commands in list of strings param filename: file name to be executed as main hydra module return: new commands to be executed in list of strings r)rrs rfun5IntegrationTestSuite.generate_custom_cmd..funs Jr )rr)r!rs rgenerate_custom_cmd(IntegrationTestSuite.generate_custom_cmds+  T#Y # $s)  r z/task_config, overrides, filename, expected_nameNz no_config.py no_config)rzhydra.job.name=overridden_nameoverridden_namedifferent_filenamerrrname_from_config_filezwith_config.py0different_filename_and_config_file_name_overrideJdifferent_filename_and_config_file_name_override_and_command_line_override task_configr&r expected_nametask_launcher_cfg extra_flagsc :Xs-n[R"U=(d 05n[R"U=(d 05n[R"Xb5n[U[5(de[ UR U5UUSUUUR5S9 g)NzHydraConfig.get().job.name)r'rr&printsexpected_outputsrr)r r[merge isinstancer rrr) r!r'rr&rrrrr_s rtest_custom_task_name*IntegrationTestSuite.test_custom_task_namesJ + %,,->-D"E&&{'8b9 oo/=#z****,,V4/* $ 8 8 : r z$task_config, overrides, expected_dirrbtask_cfgztask_cfg_${hydra.job.num})dirsubdirztask_cfg/task_cfg_0sweep_dir_config_overridehydra.sweep.dir=cli_dir+hydra.sweep.subdir=cli_dir_${hydra.job.num}cli_dir/cli_dir_0sweep_dir_cli_override sweep_dir_cli_overridding_configrz${hydra.job.override_dirname}hellorY)rrba=1b=2zhydra_cfg/a=1,b=2zsubdir:override_dirnameconfigoverride_dirnamer+seed)kv_sepitem_sep exclude_keys)rbrz???)rrrr)rrzseed=10zhydra_cfg/a_1+b_2zsubdir:custom_override_dirname expected_dirc Xc-n[R"U=(d 05n[R"U=(d 05n[R"XR5n[U[5(deUR 5nU(a[ URU5U- 5O[ URU5U- 5n [URU5UUS/-SU UR5S9 g)Nhydra.job.chdir=True os.getcwd()r'rr&rrr) r r[rrr rrrrr) r!r'rr&rrrr_ test_app_dirrs rtest_custom_sweeper_run_workdir4IntegrationTestSuite.test_custom_sweeper_run_workdirsz + %,,->-D"E&&{'8b9 oo/=#z****446  )),7,F GT..v6EF  ,,V4#9":: - $ 8 8 :  r c Un[R"U=(d 05n[R"5n[R"X%5n[U[5(de[ UR U5UUS[RR[UR U555UR5S9 g)Nzhydra.utils.get_original_cwd()r) r r[rrr rrosrrealpathrr)r!r'rrr&rr_s rtest_get_orig_dir_multirun/IntegrationTestSuite.test_get_orig_dir_multirunts %,,->-D"E&&( oo/=#z****,,V43WW--c$2K2KF2S.TU $ 8 8 :  r c SnU/SQ-n[R"U=(d 05n[R"5n[R"X&5n[U[5(de[ [ S5R55n[URU5UUSUUR5S9 UR5n U (aU OUn [RR[ URU5S- 55[ URU 5U- 5/n [URU5UUSS/U UR5S9 g) Nr)rrrz/foo/barz(hydra.utils.to_absolute_path('/foo/bar')rzfoo/barz'hydra.utils.to_absolute_path('foo/bar')r)r r[rrr rrabsoluterrrrrrr) r!r'rrrr&rr_rrroutputss rtest_to_absolute_path_multirun3IntegrationTestSuite.test_to_absolute_path_multiruns=+ #  &,,->-D"E&&( oo/=#z****4 #,,./,,V4=! $ 8 8 :  446 &2l  GG  S!:!:6!BY!NO P ))+6E F  ,,V4=}M$ $ 8 8 :  r r)rrrrrrrrrrrrr parametrizer r rrrrrrr rrrs"(4.4DXc49n%=  9 $NKK H 12!'   56+B"CDE 'E   56+B"CDE12 !_  ! 6    9      & #Y   76 4 . #-&A"%.  -A$+  #-&A".A$5   #.&E" ! #,   $/&E" % 2.1035;H5"'  !!%(*#31 mP Sh   9    &  #Y   iSh <  /9 HLS   $$ $ /9$ HLS $ $ r r)#__doc__rrrCpathlibrtypingrrrrr omegaconfr r pytestr r r rr hydra.errorsrhydra.test_utils.test_utilsrrr usefixturesrrrr/r5rOrrr rr sk 55+&&',-Y4Y4.Y4x,-==.=@0$0Cy0L)0 0  0B1J$1JCy1JL)1J 1J  1Jh ;$ ;Cy ;L) ; ;  ;F,-| | .| r