Skip to content

Commit b1da4a8

Browse files
committed
remove remaining special cases for n_rep == 1
1 parent 305a2e3 commit b1da4a8

File tree

1 file changed

+2
-15
lines changed

1 file changed

+2
-15
lines changed

doubleml/utils/blp.py

Lines changed: 2 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -73,29 +73,18 @@ def __str__(self):
7373
@property
7474
def blp_model(self):
7575
"""
76-
Best-Linear-Predictor model.
77-
For multiple repetitions this is a list with one model per repetition.
76+
Best-Linear-Predictor models.
77+
This is a list with one model per repetition.
7878
"""
7979
if self._blp_model is None:
8080
return None
81-
if self.n_rep == 1:
82-
return self._blp_model[0]
83-
return self._blp_model
84-
85-
@property
86-
def blp_models(self):
87-
"""
88-
Best-Linear-Predictor models for each repetition.
89-
"""
9081
return self._blp_model
9182

9283
@property
9384
def orth_signal(self):
9485
"""
9586
Orthogonal signal.
9687
"""
97-
if self.n_rep == 1:
98-
return self._orth_signal.reshape(-1)
9988
return self._orth_signal
10089

10190
@property
@@ -120,8 +109,6 @@ def blp_omega(self):
120109
"""
121110
if self._blp_omega is None:
122111
return None
123-
if self.n_rep == 1:
124-
return self._blp_omega[:, :, 0]
125112
return self._blp_omega
126113

127114
@property

0 commit comments

Comments
 (0)