Skip to content
Snippets Groups Projects

allow class methods to be passed to hyper model

Merged Colm Talbot requested to merge fix-hyper-model into master
1 file
+ 5
0
Compare changes
  • Side-by-side
  • Inline
+ 5
0
@@ -30,6 +30,11 @@ class Model(object):
return probability
def _get_function_parameters(self, func):
"""If the function is a class method we need to remove more arguments"""
param_keys = infer_parameters_from_function(func)
ignore = ['dataset', 'self', 'cls']
for key in ignore:
if key in param_keys:
del param_keys[param_keys.index(key)]
parameters = {key: self.parameters[key] for key in param_keys}
return parameters
Loading