Calculate number of point in a radius

def calc_poi_avg_in_r(poi, entity, radius, **kwargs):
    entity = entity.copy()
    KDT = KDTree(poi[['lat', 'long']])
    deg_to_m = 1/108000.0
    query = entity.progress_apply(lambda g: KDT.query_radius([[g.lat, g.long1]], r= deg_to_m * radius), axis=1)
    name = kwargs['name'] if kwargs['name'] else 'default'
    entity.loc[:, f'{name}'] = [ len(x[0]) for x in tqdm(query)]
    return entity