AIS Fixes

I fixed a problem in `dbm_metrics.py` where the script would fail to work on CPU. I also brought back the `test_dbm_metrics.py` test into the test suite.