Add an option to insert 2 datasets - train and test. fill_na_by_ratio(train, test=None, column_name) if test = None: - fill Na in the train set by the train col ratio. else: - fill Na in the **test** set by the **train** col ratio.