Should treatment and control size be switched when using get_treat_size?

[bpark738]

In the utils, get_treat_size function is defined to be

def get_treat_size(t, treat_split=0.5):

num_treatment = t[t > treat_split].shape[0]
num_control = t[t <= treat_split].shape[0]

return num_treatment, num_control

But in the CTL.py code , it looks like its used like this:

tb_num_cont, tb_num_treat = get_treat_size(train_treat1)

where I think cont indicates control and treat indicates treated.

Shouldn't it be

tb_num_treat, tb_num_cont= get_treat_size(train_treat1)?