def initialize()

in deploy-models/real-time-inference/multi_model_catboost/container/model_handler.py [0:0]

• 20 lines of code
• 4 McCabe index (conditional complexity)

    def initialize(self, ctx):
        start = time.time()
        self.device = "cpu"

        properties = ctx.system_properties
        self.device = "cpu"
        model_dir = properties.get("model_dir")

        print("model_dir {}".format(model_dir))
        print(os.system("ls {}".format(model_dir)))

        model_file = CatBoostClassifier()

        onlyfiles = [
            f
            for f in os.listdir(model_dir)
            if os.path.isfile(os.path.join(model_dir, f)) and f.endswith(".bin")
        ]
        print(
            f"Modelhandler:model_file location::{model_dir}:: files:bin:={onlyfiles} :: going to load the first one::"
        )
        self.model = model_file = model_file.load_model(onlyfiles[0])

        self.initialized = True
        print(f" perf initialize {(time.time() - start) * 1000} ms")