def load_context()

in Project-AutoML/predictor.py [0:0]

• 8 lines of code
• 3 McCabe index (conditional complexity)

    def load_context(self, context):
        model_path = context.artifacts["model_path"]
        if "autosklearn" in model_path:
            self.predictor = self.load_autosklearn_predictor(model_path)

        elif "flaml" in model_path:
            self.predictor = self.load_flaml_predictor(model_path)

        else:
            assert f"cant not load model from path {model_path}"