o ziW'@sddlmZmZmZmZddlmZddlmZ ddl m Z ddl m Z ddlmZddlmZedZdd d d d d d d d d d d d d d d d d ed eedeedeedededeededededededededededeeeeeefff"ddZd S))DictListOptionalTuple)RequirementCache)_runif_reasons)_PSUTIL_AVAILABLE)_RICH_AVAILABLE)_ONNX_AVAILABLE)_OMEGACONF_AVAILABLE scikit-learnNF) min_cuda_gpus min_torch max_torch min_python bf16_cudatpumps skip_windows standalone deepspeeddynamorich omegaconfpsutilsklearnonnxr rrrrrrrrrrrrrrrreturnc Cst|||||||||| | d \}}| rts|d| r#ts#|d| r,ts,|d|r5ts5|d|r>ts>|d||fS)aConstruct reasons for pytest skipif. Args: min_cuda_gpus: Require this number of gpus and that the ``PL_RUN_CUDA_TESTS=1`` environment variable is set. min_torch: Require that PyTorch is greater or equal than this version. max_torch: Require that PyTorch is less than this version. min_python: Require that Python is greater or equal than this version. bf16_cuda: Require that CUDA device supports bf16. tpu: Require that TPU is available. mps: If True: Require that MPS (Apple Silicon) is available, if False: Explicitly Require that MPS is not available skip_windows: Skip for Windows platform. standalone: Mark the test as standalone, our CI will run it in a separate process. This requires that the ``PL_RUN_STANDALONE_TESTS=1`` environment variable is set. deepspeed: Require that microsoft/DeepSpeed is installed. dynamo: Require that `torch.dynamo` is supported. rich: Require that willmcgugan/rich is installed. omegaconf: Require that omry/omegaconf is installed. psutil: Require that psutil is installed. sklearn: Require that scikit-learn is installed. onnx: Require that onnx is installed. ) r rrrrrrrrrrRichrrr r) fabric_run_ifr appendr r_SKLEARN_AVAILABLEr )r rrrrrrrrrrrrrrrreasonskwargsr$^/home/ubuntu/.local/lib/python3.10/site-packages/pytorch_lightning/utilities/testing/_runif.pyrs0+      r)typingrrrr lightning_utilities.core.importsr"lightning_fabric.utilities.testingrr"pytorch_lightning.accelerators.cpur2pytorch_lightning.callbacks.progress.rich_progressr pytorch_lightning.core.moduler #pytorch_lightning.utilities.importsr r!intstrboolr$r$r$r%sv