Skip to content

Commit 6388b11

Browse files
committed
docstrings
1 parent 9b7a575 commit 6388b11

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

pyfixest/estimation/cupy/demean_cupy_.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -144,20 +144,19 @@ def demean(
144144
145145
Parameters
146146
----------
147-
x : np.ndarray, shape (n_obs,) or (n_obs, n_vars)
147+
x : np.ndarray.
148148
Variable(s) to demean.
149-
flist : np.ndarray, shape (n_obs, n_factors) or (n_obs,)
149+
flist : np.ndarray.
150150
Integer-encoded fixed effects. Ignored if fe_sparse_matrix provided.
151+
Usually not used within pyfixest internals.
151152
weights : np.ndarray, shape (n_obs,)
152153
Weights (1.0 for equal weighting).
153154
tol : float, default=1e-8
154-
Convergence tolerance (unused for normal eq; used for LSMR fallback).
155+
Convergence tolerance. Used for both atol and btol of lsmr algo.
155156
maxiter : int, default=100_000
156-
Maximum iterations (used for LSMR fallback).
157+
Maximum iterations for lsmr iterations.
157158
fe_sparse_matrix : scipy.sparse.csr_matrix, optional
158-
Pre-computed sparse FE dummy matrix (from formulaic).
159-
If provided, uses this directly (faster and more accurate).
160-
Should have reference levels dropped for full-rank system.
159+
Pre-computed sparse FE dummy matrix.
161160
162161
Returns
163162
-------

0 commit comments

Comments
 (0)