diff options
Diffstat (limited to 'lib/parameters.py')
| -rw-r--r-- | lib/parameters.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/parameters.py b/lib/parameters.py index a636b52..207b8d8 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -606,7 +606,7 @@ class ModelAttribute: "argCount": self.arg_count, "modelFunction": self.model_function.to_json(**kwargs), } - if type(self.model_function) == df.CARTFunction: + if type(self.model_function) in (df.CARTFunction, df.FOLFunction): feature_names = self.param_names feature_names += list( map( |