diff --git a/clarifai/runners/models/model_builder.py b/clarifai/runners/models/model_builder.py index 11e41fe5..1d988cf8 100644 --- a/clarifai/runners/models/model_builder.py +++ b/clarifai/runners/models/model_builder.py @@ -792,7 +792,9 @@ def get_method_signatures(self, mocking=True): """ model_class = self.load_model_class(mocking=mocking) method_infos = model_class._get_method_infos() - signatures = [method.signature for method in method_infos.values()] + signatures = [ + method.signature for method in method_infos.values() if method.signature is not None + ] return signatures @property