data_utils/feature_importance_simulation.py

import pandas as pd
import numpy as np
from data_utils.data_simulation import UpliftSimulationReady

class FISimulationReady:

    def __init__(self, files_path):
        self.files_path = files_path

    def load_feature_importance(self, file_name):
        uplift_simulation = UpliftSimulationReady(self.files_path)
        df = uplift_simulation.load_uplift_data(file_name)
        return df