o xi1@sdZddlmZddlZddlmZddlZddlmZddl m Z ddl m Z Gdd d ej ZedRd d ZejedddejdSddZeejddejddddejdddddedddTdUd#dZedRd$d%Zeejd&ejddd'd(ejd)ddd*d+ejd,d-dd.d/ejd0d1dd2d/ejd3d4dd5d/ejd6dd7d/ejd8dd9d/ejd:ddd;d+ejdd?ddd@d+ejdAdBdCd/dVdPdQZdS)WzwBeta versions of wandb CLI commands. These commands are experimental and may change or be removed in future versions. ) annotationsN)Any) get_sentry)WandbCoreNotAvailableError) get_core_pathcs@eZdZdZdddfd d Zdfdd ZdddZZS)DefaultCommandGroupa A click Group that falls through to a default command. If the first argument isn't a recognized subcommand, the default command is invoked with all arguments passed through. This allows backward-compatible CLIs where `cmd [path]` and `cmd run [path]` are equivalent. run) default_cmdargsrr strkwargsreturnNonecstj|i|||_dS)N)super__init__r )selfr r r  __class__B/home/ubuntu/.local/lib/python3.10/site-packages/wandb/cli/beta.pyrs zDefaultCommandGroup.__init__ctx click.Context list[str]cs:|r|dds|d|jvr|jg|}t||S)Nr-) startswithcommandsr r parse_args)rrr rrrrs  zDefaultCommandGroup.parse_args formatterclick.HelpFormattercCs||jddS)Nz[PATH] | COMMAND [ARGS]...) write_usage command_path)rrrrrr format_usage$sz DefaultCommandGroup.format_usage)r rr r r rr r)rrr rr r)rrrrr r)__name__ __module__ __qualname____doc__rrr! __classcell__rrrrrs rr rc CsdtjddztWdSty1}ztd|tj|dddWYd}~dSd}~ww)zzBeta versions of wandb CLI commands. These commands may change or even completely break in any release of wandb. wandb_beta)process_contextz using `wandb beta`. failed with redT)fgerrN)rconfigure_scoperr exceptionclicksecho)errrbeta(s r1rT)clsr invoke_without_commandrrcCsdS)zW&B LEET: the Lightweight Experiment Exploration Tool. A terminal UI for viewing your W&B runs locally. Examples: wandb beta leet View latest run wandb beta leet ./wandb View runs in directory Nr)rrrrleet;s r4path)existsF)nargstyperequiredz--pprofz;Serve /debug/pprof/* on this address (e.g. 127.0.0.1:6060).)defaulthiddenhelpz-hz--help str | Nonepprofr cCsddlm}|||dS)zLaunch the LEET TUI. PATH can be a .wandb file, a run directory, or a wandb directory. If omitted, searches for the latest run. r6 beta_leetN)r;rBlaunch)r5r@rBrrrrIs cCsddlm}|dS)zEdit LEET configuration.r6rAN)r;rB launch_configrArrrconfig]s  rEpaths)r9r8z--livezvSync a run while it's still being logged. This may hang if the process generating the run crashes uncleanly. )is_flagr<r>z-ez--entityz4An entity override to use for all runs being synced.)r<r>z-pz --projectz4A project override to use for all runs being synced.z--idrun_idaA run ID override to use for all runs being synced. If setting this and syncing multiple files (with the same entity and project), the files will be synced in order of start time. This is intended to work with syncing multiple resumed fragments of the same run. z --job-typez.A job type override for all runs being synced.z--replace-tagsz3Rename tags using the format 'old1=new1,old2=new2'.z--skip-synced/--no-skip-syncedz:Skip runs that have already been synced with this command.z --dry-runz3Print what would happen without uploading anything.z-vz --verbosezPrint more information.z-na Max number of runs to sync at a time. When syncing multiple files that are part of the same run, the files are synced sequentially in order of start time regardless of this setting. This happens for resumed runs or when using the --id parameter. tuple[str, ...]liveboolentityprojectjob_type replace_tags skip_synceddry_runverbosenintc Cs:ddlm} | jdd|D||||||||| | d dS)aUpload .wandb files specified by PATHS. This is a beta re-implementation of `wandb sync`. It is not feature complete, not guaranteed to work, and may change in backward-incompatible ways in any release of wandb. PATHS can include .wandb files, run directories containing .wandb files, and "wandb" directories containing run directories. For example, to sync all runs in a directory: wandb beta sync ./wandb To sync a specific run: wandb beta sync ./wandb/run-20250813_124246-n67z9ude Or equivalently: wandb beta sync ./wandb/run-20250813_124246-n67z9ude/run-n67z9ude.wandb r6) beta_synccSsg|]}t|qSr)pathlibPath).0r5rrr szsync..) rLrNrOrIrPrQrSrRrT parallelismN)r;rWsync) rFrLrNrOrIrPrQrRrSrTrUrWrrrr]es m  r])r r)rrr r)Nr;)r5r?r@r r r)rFrKrLrMrNr rOr rIr rPr rQr rRrMrSrMrTrMrUrVr r)r% __future__rrXtypingrr.wandb.analyticsr wandb.errorsr wandb.utilrGrouprgroupr1 pass_contextr4commandargumentrYoption help_optionrrEr]rrrrs           &