Missing import in visualisation.py

[nhsavage]

Flake8 shows up this error: visualisation.py:146:28: F821 undefined name 'np'

1. need to check if this section of code is tested by unit tests. Ideally make a failing test
2. fix the test failure
3. remove F821 from the ignored warnings