Skip to content
Snippets Groups Projects
Commit 6a25ae13 authored by Colm Talbot's avatar Colm Talbot
Browse files

allow class methods to be passed to hyper model

parent 060da071
No related branches found
No related tags found
No related merge requests found
......@@ -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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment