[BUG]: phishing validation script fails on labels assertion

[dagardner-nv]

Version

23.11

Which installation method(s) does this occur on?

Source

Describe the bug.

    assert len(self._idx2label) > 0, "No labels were added to the stage"
AssertionError: No labels were added to the stage

Minimum reproducible example

./scripts/validation/phishing/val-phishing-all.sh

Relevant log output

Click here to see error details

[Paste the error here, it will be hidden by default]

Full env printout

Click here to see environment details

Launching Triton Container

Triton is ready.

Model 'phishing-bert-onnx' loaded in Triton

Configuring Pipeline via CLI

Loaded labels file. Current labels: [['not_phishing', 'is_phishing']]

The label 'pred' is not in Config.class_labels and will be ignored

Traceback (most recent call last):

File "/home/dagardner/work/conda/envs/morpheus/bin/morpheus", line 8, in 

sys.exit(run_cli())

File "/home/dagardner/work/morpheus/morpheus/cli/run.py", line 20, in run_cli

cli(obj={}, auto_envvar_prefix='MORPHEUS', show_default=True, prog_name="morpheus")

File "/home/dagardner/work/conda/envs/morpheus/lib/python3.10/site-packages/click/core.py", line 1157, in call

return self.main(*args, **kwargs)

File "/home/dagardner/work/conda/envs/morpheus/lib/python3.10/site-packages/click/core.py", line 1078, in main

rv = self.invoke(ctx)

File "/home/dagardner/work/conda/envs/morpheus/lib/python3.10/site-packages/click/core.py", line 1688, in invoke

return _process_result(sub_ctx.command.invoke(sub_ctx))

File "/home/dagardner/work/conda/envs/morpheus/lib/python3.10/site-packages/click/core.py", line 1688, in invoke

return _process_result(sub_ctx.command.invoke(sub_ctx))

File "/home/dagardner/work/conda/envs/morpheus/lib/python3.10/site-packages/click/core.py", line 1719, in invoke

rv.append(sub_ctx.command.invoke(sub_ctx))

File "/home/dagardner/work/conda/envs/morpheus/lib/python3.10/site-packages/click/core.py", line 1434, in invoke

return ctx.invoke(self.callback, **ctx.params)

File "/home/dagardner/work/conda/envs/morpheus/lib/python3.10/site-packages/click/core.py", line 783, in invoke

return __callback(*args, **kwargs)

File "/home/dagardner/work/morpheus/morpheus/cli/utils.py", line 114, in new_func

return f(ctx, *args, **kwargs)

File "/home/dagardner/work/morpheus/morpheus/cli/register_stage.py", line 352, in command_callback

stage = stage_class(**kwargs)

File "/home/dagardner/work/morpheus/morpheus/pipeline/stage_base.py", line 44, in inner

func(self, *args, **kwargs)

File "/home/dagardner/work/morpheus/morpheus/stages/postprocess/add_classifications_stage.py", line 60, in init

super().init(c, labels=labels, prefix=prefix, probs_type=probs_type, threshold=threshold)

File "/home/dagardner/work/morpheus/morpheus/pipeline/stage_base.py", line 44, in inner

func(self, *args, **kwargs)

File "/home/dagardner/work/morpheus/morpheus/stages/postprocess/add_scores_stage_base.py", line 81, in init

assert len(self._idx2label) > 0, "No labels were added to the stage"

AssertionError: No labels were added to the stage

Killing Triton container

85a81858845cbb3013dbf733b1e429b34525cb20f58b6615d43daaa76bfee521

Other/Misc.

No response

Code of Conduct

• I agree to follow Morpheus' Code of Conduct
• I have searched the open bugs and have found no duplicates for this bug report