File tree Expand file tree Collapse file tree 2 files changed +14
-8
lines changed
Expand file tree Collapse file tree 2 files changed +14
-8
lines changed Original file line number Diff line number Diff line change @@ -520,11 +520,14 @@ 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__ )
523+ spec = list (
524+ inspect .signature (skllearner .__init__ ).parameters .keys ()
525+ )
524526 # 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 }
527+ assert spec [0 ] == "self"
528+ params = {
529+ name : values [name ] for name in spec [1 :] if name in values
530+ }
528531 else :
529532 raise TypeError ("Wrapper does not define '__wraps__'" )
530533 return params
Original file line number Diff line number Diff line change @@ -196,11 +196,14 @@ 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__ )
199+ spec = list (
200+ inspect .signature (sklprojection .__init__ ).parameters .keys ()
201+ )
200202 # 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 }
203+ assert spec [0 ] == "self"
204+ params = {
205+ name : values [name ] for name in spec [1 :] if name in values
206+ }
204207 else :
205208 raise TypeError ("Wrapper does not define '__wraps__'" )
206209 return params
You can’t perform that action at this time.
0 commit comments