Fix coverage in generation_strategy.py

mpolson64 · facebook-github-bot · commit a67302f28a99 · 2022-11-07T13:57:37.000-08:00
Summary: As titled

Reviewed By: bernardbeckerman

Differential Revision: D41088315

fbshipit-source-id: 93f513f733c67e165d6dd5f59b8db79f827c509e
diff --git a/ax/modelbridge/generation_strategy.py b/ax/modelbridge/generation_strategy.py
@@ -601,7 +601,7 @@ def _maybe_move_to_next_step(self, raise_data_required_error: bool = True) -> bo
                 ]
             ):
                 if len(self._steps) == self._curr.index + 1:
-                    raise GenerationStrategyCompleted(
+                    raise GenerationStrategyCompleted(  # pragma: no cover
                         f"Generation strategy {self} generated all the trials as "
                         "specified in its steps."
                     )

Original file line number	Diff line number	Diff line change
`@@ -601,7 +601,7 @@ def _maybe_move_to_next_step(self, raise_data_required_error: bool = True) -> bo`
`601`	`601`	`]`
`602`	`602`	`):`
`603`	`603`	`if len(self._steps) == self._curr.index + 1:`
`604`		`- raise GenerationStrategyCompleted(`
	`604`	`+ raise GenerationStrategyCompleted( # pragma: no cover`
`605`	`605`	`f"Generation strategy {self} generated all the trials as "`
`606`	`606`	`"specified in its steps."`
`607`	`607`	`)`