tniIdZddlZddlZddlmZmZddlmZmZm Z m Z ej Z e j ZdefdZdeeefddfdZdeeefddfd ZdS) zz runpod | serverless | worker_loop.py Called to convert a container into a worker pod for the runpod serverless platform. N)AnyDict) rp_loggerrp_localrp_pingrp_scalereturnc|dddrdStjdddSdS)z?Returns True if the worker is running locally, False otherwise.rp_args test_inputNTRUNPOD_WEBHOOK_GET_JOBF)getosenvironconfigs U/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/runpod/serverless/worker.py _is_localrsE i\400t z~~.55=t 5rcttj|}|dS)a! Starts the worker loop for multi-processing. This function is called when the worker is running on Runpod. This function starts a loop that runs indefinitely until the worker is killed. Args: config (Dict[str, Any]): Configuration parameters for the worker. N) heartbeat start_pingr JobScalerstart)r job_scalers r run_workerrs@#F++Jrct|r(tjtj|dSt |dS)z Checks if the worker is running locally or on Runpod. If running locally, the test job is run and the worker exits. If running on Runpod, the worker loop is created. N)rasynciorunr run_localrrs rmainr!.sK  H&v..///// 6r)__doc__rrtypingrrrunpod.serverless.modulesrrrr RunPodLoggerlog Heartbeatrboolrstrrr!rrr+s  LLLLLLLLLLLLi G    tCH~$$ c3h D      r