remove pymorch dependency