Update snap.py

removed unused import