Skip to content

Commit

Permalink
Modify error msg
Browse files Browse the repository at this point in the history
  • Loading branch information
FrancescMartiEscofetQC committed Jul 15, 2024
1 parent 4e9b975 commit 249a583
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions metalearners/slearner.py
Original file line number Diff line number Diff line change
Expand Up @@ -289,12 +289,10 @@ def predict_conditional_average_outcomes(
@classmethod
def necessary_onnx_models(cls) -> set[str]:
raise ValueError(
"The SLearner does not implement this method. Please refer to comment in the tutorial "
"about how to do this."
"The SLearner does not implement this method. Please refer to comment in the tutorial."
)

def build_onnx(self, models: Mapping[str, Sequence], output_name: str = "tau"):
raise ValueError(
"The SLearner does not implement this method. Please refer to the tutorial "
"on the documentation on how to do this."
"The SLearner does not implement this method. Please refer to comment in the tutorial."
)

0 comments on commit 249a583

Please sign in to comment.