Skip to content

Commit

Permalink
Fix typo.
Browse files Browse the repository at this point in the history
  • Loading branch information
kklein committed Aug 9, 2024
1 parent ce2334b commit 5701248
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions metalearners/rlearner.py
Original file line number Diff line number Diff line change
Expand Up @@ -586,8 +586,8 @@ def predict_conditional_average_outcomes(

control_outcomes = outcome_estimates

# TODO: Consider whether the readability vs efficiency trade-off should be dealth with differently here.
# One could use a matrix/tensor operation.
# TODO: Consider whether the readability vs efficiency trade-off should be dealt with differently here.
# One could use matrix/tensor operations instead.
for treatment_variant in range(1, self.n_variants):
if (n_outputs := cate_estimates.shape[2]) > 1:
for outcome_channel in range(0, n_outputs):
Expand Down

0 comments on commit 5701248

Please sign in to comment.