Skip to content

Commit

Permalink
fix linting
Browse files Browse the repository at this point in the history
  • Loading branch information
fverac committed Apr 5, 2022
1 parent 73c492a commit 401cf37
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion econml/sklearn_extensions/linear_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -1684,7 +1684,7 @@ def _check_input(self, X, y, sample_weight, freq_weight, sample_var):
"""Check dimensions and other assertions."""

X, y, sample_weight, freq_weight, sample_var = check_input_arrays(
X, y, sample_weight, freq_weight, sample_var, dtype = 'numeric')
X, y, sample_weight, freq_weight, sample_var, dtype='numeric')
if X is None:
X = np.empty((y.shape[0], 0))
if self.fit_intercept:
Expand Down
2 changes: 1 addition & 1 deletion econml/tests/test_statsmodels.py
Original file line number Diff line number Diff line change
Expand Up @@ -333,7 +333,7 @@ def test_o_dtype(self):
np.random.seed(123)
n = 1000
d = 3

X = np.random.normal(size=(n, d)).astype('O')
y = np.random.normal(size=n).astype('O')

Expand Down
2 changes: 1 addition & 1 deletion econml/utilities.py
Original file line number Diff line number Diff line change
Expand Up @@ -532,7 +532,7 @@ def check_input_arrays(*args, validate_len=True, force_all_finite=True, dtype=No
Whether to allow inf and nan in input arrays.
dtype : 'numeric', type, list of type or None (default=None)
Argument passed to sklearn.utils.check_array.
Argument passed to sklearn.utils.check_array.
Specifies data type of result. If None, the dtype of the input is preserved.
If "numeric", dtype is preserved unless array.dtype is object.
If dtype is a list of types, conversion on the first type is only
Expand Down

0 comments on commit 401cf37

Please sign in to comment.