Update ErbiumAmplifier.py

Imports are now written in methods where they're used