Skip to content

Commit df3b613

Browse files
test: remove redundant output_kind override test
1 parent 45e5892 commit df3b613

File tree

1 file changed

+0
-20
lines changed

1 file changed

+0
-20
lines changed

tests/unit_tests/test_generator_config.py

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -132,26 +132,6 @@ def test_generator_yaml_config_loading(self):
132132
self.assertEqual(generator.sampling_params.n, 2)
133133
self.assertEqual(generator.sampling_params.max_tokens, 32)
134134

135-
@pytest.mark.skipif(
136-
_import_error(),
137-
reason="Import error, likely due to missing dependencies on CI.",
138-
)
139-
def test_generator_overrides_output_kind_to_final_only(self):
140-
"""Generator overrides output_kind to FINAL_ONLY when initialized with dict."""
141-
from forge.actors.generator import Generator
142-
from vllm.sampling_params import RequestOutputKind
143-
144-
sampling_dict = {
145-
"n": 1,
146-
"max_tokens": 16,
147-
"output_kind": RequestOutputKind.DELTA,
148-
}
149-
150-
generator = Generator(sampling_params=sampling_dict)
151-
self.assertEqual(
152-
generator.sampling_params.output_kind, RequestOutputKind.FINAL_ONLY
153-
)
154-
155135

156136
if __name__ == "__main__":
157137
unittest.main()

0 commit comments

Comments
 (0)