Skip to content

Commit efaa63b

Browse files
committed
add warning for setting kwargs
1 parent 1566af4 commit efaa63b

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

doubleml/rdd/rdd.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,10 @@ def __init__(
143143

144144
self._check_effect_sign()
145145

146-
# TODO: Add further input checks
146+
if ("h", "b") & kwargs.keys():
147+
warnings.warn(f"Key-worded arguments contain: {('h', 'b') & kwargs.keys()}. \n \
148+
Iterative bandwidth selection will be overwritten by provided values.")
149+
147150
self.kwargs = kwargs
148151

149152
self._smpls = DoubleMLResampling(
@@ -454,11 +457,11 @@ def _fit_rdd(self, h=None, b=None):
454457
if self.fuzzy:
455458
rdd_res = rdrobust.rdrobust(
456459
y=self._M_Y[:, self._i_rep], x=self._score, fuzzy=self._M_D[:, self._i_rep],
457-
c=0, **{{"h": h, "b": b} | self.kwargs}
460+
c=0, **({"h": h, "b": b} | self.kwargs)
458461
)
459462
else:
460463
rdd_res = rdrobust.rdrobust(y=self._M_Y[:, self._i_rep], x=self._score, fuzzy=None,
461-
c=0, **{{"h": h, "b": b} | self.kwargs})
464+
c=0, **({"h": h, "b": b} | self.kwargs))
462465
return rdd_res
463466

464467
def _set_coefs(self, rdd_res, h):

0 commit comments

Comments
 (0)