o xi@sdZddlmZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z ddlZddlmZddlmZdd lmZeeZe d ZGd d d ZGd dde ZGddde ZdS)zImplementation of the abstract runner class. This class defines the interface that the W&B launch runner uses to manage the lifecycle of runs launched in different environments (e.g. runs launched locally or in a cluster). ) annotationsN)ABCabstractmethod)AnyLiteral)Api)runid) LaunchProject)unknownstartingrunningfailedfinishedstoppingstopped preemptedc@s@eZdZddddZdd d Zdd dZdddZdddZdS)Statusr NstateStatemessages list[str]cCs||_|pg|_dSN)rr)selfrrrT/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/launch/runner/abstract.py__init__'szStatus.__init__returncC|jSrrrrrr__repr__+zStatus.__repr__strcCrrrr rrr__str__.r"zStatus.__str___Status__valueobjectboolcCs t|tr |j|jkS|j|kSr) isinstancerr)rr%rrr__eq__1s   z Status.__eq__intcCs t|jSr)hashrr rrr__hash__7s zStatus.__hash__)r N)rrrr)rr)rr#)r%r&rr')rr*)__name__ __module__ __qualname__rr!r$r)r,rrrrr&s     rc@seZdZdZdddZedddZed d d Z d!d"ddZ ed#ddZ edddZ edddZ eed ddZ dS)$ AbstractRuna&Wrapper around a W&B launch run. A launched run is a subprocess running an entry point command, that exposes methods for waiting on and cancelling the run. This class defines the interface that the W&B launch runner uses to manage the lifecycle of runs launched in different environments (e.g. runs launched locally or in a cluster). ``AbstractRun`` is not thread-safe. That is, concurrent calls to wait() / cancel() from multiple threads may inadvertently kill resources (e.g. local processes) unrelated to the run. rNonecCs t|_dSr)r_statusr rrrrGs zAbstractRun.__init__rcCrr)r2r rrrstatusJszAbstractRun.status str | NonecdS)z(Return the logs associated with the run.Nrr rrrget_logsNzAbstractRun.get_logsFcmdr output_only bool | None&subprocess.Popen[bytes] | bytes | Nonec Cszz tj}tj||tjd}|r||jdur|jWS|WStjy<}zt d|WYd}~dSd}~ww)zRun the command and returns a popen object or the stdout of the command. Arguments: cmd: The command to run output_only: If true just return the stdout bytes )envstdoutNzCommand failed: ) osenviron subprocessPopenPIPEwaitr=readCalledProcessErrorwandb termerror)rr8r9r<popenerrr_run_cmdRs   zAbstractRun._run_cmdr'cr5)zWait for the run to finish, returning True if the run succeeded and false otherwise. Note that in some cases, we may wait until the remote job completes rather than until the W&B run completes. Nrr rrrrCgr7zAbstractRun.waitcr5)zGet status of the run.Nrr rrr get_statusnr7zAbstractRun.get_statuscr5)zCancel the run (interrupts the command subprocess, cancels the run, etc). Cancels the run and waits for it to terminate. The W&B run status may not be set correctly upon run cancellation. Nrr rrrcancelrr7zAbstractRun.cancelcCsdSrrr rrridzszAbstractRun.idN)rr1)rr)rr4)F)r8rr9r:rr;rr')r-r.r/__doc__rpropertyr3rr6rJrCrKrLrMrrrrr0;s$       r0c@sTeZdZUdZded<dd d ZdddZedddZd ddZ e d!ddZ dS)"AbstractRunneraDAbstract plugin class defining the interface needed to execute W&B Launches. You can define subclasses of ``AbstractRunner`` and expose them as third-party plugins to enable running W&B projects against custom execution backends (e.g. to run projects against your team's in-house cluster or job scheduler). r#_typeapirbackend_configdict[str, Any]rr1cCs$||_||_t|_t|_dSr)_apirTr>getcwd_cwdr generate_id _namespace)rrSrTrrrrs zAbstractRunner.__init__r8r4cCs t|S)z>Cross platform utility for checking if a program is available.)shutilwhich)rr8rrrfind_executables zAbstractRunner.find_executablercCs|jjSr)rVapi_keyr rrrr^szAbstractRunner.api_keyr'cCs$|jjdurtdtddS)zThis is called on first boot to verify the needed commands, and permissions are available. For now just call `wandb.termerror` and `sys.exit(1)` Nz?Couldn't find W&B api key, run wandb login or set WANDB_API_KEYT)rVr^rFrGsysexitr rrrverifys  zAbstractRunner.verifylaunch_projectr image_uriAbstractRun | Nonecr5)aSubmit an LaunchProject to be run. Returns a SubmittedRun object to track the execution Arguments: launch_project: Object of _project_spec.LaunchProject class representing a wandb launch project Returns: A :py:class:`wandb.sdk.launch.runners.SubmittedRun`. This function is expected to run the project asynchronously, i.e. it should trigger project execution and then immediately return a `SubmittedRun` to track execution status. Nr)rrcrdrrrrunr7zAbstractRunner.runN)rSrrTrUrr1)r8r#rr4)rrrN)rcr rdr#rre) r-r.r/rO__annotations__rr]rPr^rbrrfrrrrrQs      rQ)rO __future__rloggingr>r[r@r`abcrrtypingrrrFwandb.apis.internalr wandb.sdk.libr _project_specr getLoggerr-_loggerrrr0rQrrrrs(      E