Update lynn_custom.py

syntax