new UserWarning introduced by torch 1.7.0

Our application started to get UserWarning from ax after the new torch 1.7.0 release

$ python example/banana.py -m 'banana.x=int(interval(-5, 5))' 'banana.y=interval(-5, 10.1)'
/private/home/jieru/.conda/envs/jh-test/lib/python3.8/site-packages/torch/cuda/__init__.py:52: UserWarning: Error instantiating/calling 'hydra_plugins.hydra_ax_sweeper.ax_sweeper.AxSweeper' : CUDA initialization: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx (Triggered internally at  /pytorch/c10/cuda/CUDAFunctions.cpp:100.)
  return torch._C._cuda_getDeviceCount() > 0

The torch warning was introduced in this torch PR and raise in ax in here

This warning is not relevant to my application and as a result it would be awesome if ax could suppress it.

[lena-kashtelyan]

This warning is now removed on master and will be removed in the imminent stable version release.

[lena-kashtelyan]

@jieru-hu, @shagunsodhani, hi! You could probably leverage the warnings library to suppress these in your application?

Something like this could work (I also advise you to look into making it more granular so you are not just suppressing any UserWarning, but that’s up to you):

import warnings
warnings.filterwarnings("ignore", category=UserWarning)

We’ll also look into what is causing these, but you should be unblocked by the above in the meantime.