File tree Expand file tree Collapse file tree 2 files changed +8
-10
lines changed
Expand file tree Collapse file tree 2 files changed +8
-10
lines changed Original file line number Diff line number Diff line change @@ -520,11 +520,10 @@ def params(self, value):
520520 def _get_sklparams (self , values ):
521521 skllearner = self .__wraps__
522522 if skllearner is not None :
523- spec = inspect .getargs (skllearner .__init__ .__code__ )
524- # first argument is 'self'
525- assert spec .args [0 ] == "self"
526- params = {name : values [name ] for name in spec .args [1 :]
527- if name in values }
523+ spec = list (
524+ inspect .signature (skllearner .__init__ ).parameters .keys ()
525+ )
526+ params = {name : values [name ] for name in spec if name in values }
528527 else :
529528 raise TypeError ("Wrapper does not define '__wraps__'" )
530529 return params
Original file line number Diff line number Diff line change @@ -196,11 +196,10 @@ def params(self, value):
196196 def _get_sklparams (self , values ):
197197 sklprojection = self .__wraps__
198198 if sklprojection is not None :
199- spec = inspect .getargs (sklprojection .__init__ .__code__ )
200- # first argument is 'self'
201- assert spec .args [0 ] == "self"
202- params = {name : values [name ] for name in spec .args [1 :]
203- if name in values }
199+ spec = list (
200+ inspect .signature (sklprojection .__init__ ).parameters .keys ()
201+ )
202+ params = {name : values [name ] for name in spec if name in values }
204203 else :
205204 raise TypeError ("Wrapper does not define '__wraps__'" )
206205 return params
You can’t perform that action at this time.
0 commit comments