Implementation of morphology models as astropy models and testing