File tree Expand file tree Collapse file tree 1 file changed +0
-20
lines changed
Expand file tree Collapse file tree 1 file changed +0
-20
lines changed Original file line number Diff line number Diff 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
156136if __name__ == "__main__" :
157137 unittest .main ()
You can’t perform that action at this time.
0 commit comments