diff --git a/examples/jax/encoder/test_multiprocessing_encoder.py b/examples/jax/encoder/test_multiprocessing_encoder.py index 5509d01e98..0d19b9943f 100644 --- a/examples/jax/encoder/test_multiprocessing_encoder.py +++ b/examples/jax/encoder/test_multiprocessing_encoder.py @@ -561,6 +561,7 @@ def query_gpu(q): num_gpus = len(subprocess.check_output(["nvidia-smi", "-L"]).decode().strip().split("\n")) q.put([num_gpus, gpu_has_fp8, gpu_has_bf16]) + def unittest_query_gpu(): r""" It is only used by TestEncoder.