Skip to content

Commit

Permalink
feat(components): use rlhf_preprocessor to replace the current value_…
Browse files Browse the repository at this point in the history
…exists call in rlhf

PiperOrigin-RevId: 616878609
  • Loading branch information
Googler committed Apr 19, 2024
1 parent 54f2e45 commit b463554
Showing 1 changed file with 3 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -191,11 +191,10 @@ def rlhf_pipeline(
encryption_spec_key_name=encryption_spec_key_name,
).set_display_name('Reinforcement Learning')

has_inference_dataset = function_based.value_exists(
value=eval_dataset
).set_display_name('Resolve Inference Dataset')
has_inference_dataset = preprocess_metadata.outputs['has_inference_dataset']

with kfp.dsl.Condition(
has_inference_dataset.output == True, # pylint: disable=singleton-comparison
has_inference_dataset == True, # pylint: disable=singleton-comparison
name='Perform Inference',
):
has_model_checkpoint = function_based.value_exists(
Expand Down

0 comments on commit b463554

Please sign in to comment.