From be57dfe1533ee1c465a9c138cb1973fb9d5ca553 Mon Sep 17 00:00:00 2001 From: sakird Date: Thu, 26 Dec 2024 14:21:59 +0300 Subject: [PATCH] remove assert that masks type error #135 --- snowflake/ml/model/_packager/model_handlers/custom.py | 1 - 1 file changed, 1 deletion(-) diff --git a/snowflake/ml/model/_packager/model_handlers/custom.py b/snowflake/ml/model/_packager/model_handlers/custom.py index 747a47c..96d11cf 100644 --- a/snowflake/ml/model/_packager/model_handlers/custom.py +++ b/snowflake/ml/model/_packager/model_handlers/custom.py @@ -98,7 +98,6 @@ def get_prediction( if model.context.model_refs: for sub_name, model_ref in model.context.model_refs.items(): handler = model_handler.find_handler(model_ref.model) - assert handler is not None if handler is None: raise TypeError("Your input type to custom model is not currently supported") sub_model = handler.cast_model(model_ref.model)