util.py

import numpy as np
from datetime import datetime
def num_features_selector(X):
    return X.select_dtypes(include=np.number).columns.to_list()   
def cat_features_selector(X):
    return X.select_dtypes(include=['object', 'category']).columns.to_list()
def store_age(establishmentYear):
    return datetime.now().year - establishmentYear