Skip to content

Commit 91414f6

Browse files
authored
Refactor test for transfer trial input validation
1 parent 3099d54 commit 91414f6

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

tests/test_trial.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@
44
from ix_function.observation import MeasuredValue, OutcomeRecord
55
from ix_function.prediction import TransferPrediction
66
from ix_function.trial import (
7-
TrialStatus,
87
TransferTrialInput,
8+
TrialStatus,
99
choose_trial_status,
1010
evaluate_bounded_claim_request,
1111
required_actions_for_trial,
@@ -78,7 +78,8 @@ def test_run_transfer_trial_blocks_failed_transfer_outcome() -> None:
7878
assert result.required_actions
7979

8080

81-
def test_validate_transfer_trial_input_blocks_prediction_intervention_mismatch() -> None:
81+
def test_validate_transfer_trial_input_blocks_prediction_intervention_mismatch(
82+
) -> None:
8283
trial_input = make_trial_input()
8384
invalid_prediction = TransferPrediction(
8485
prediction_id=trial_input.prediction.prediction_id,

0 commit comments

Comments
 (0)