Skip to content

Commit

Permalink
Fix parameter validation for sources
Browse files Browse the repository at this point in the history
  • Loading branch information
dagardner-nv committed Aug 30, 2024
1 parent 7ee095b commit 1c82682
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions python/morpheus/morpheus/pipeline/stage_decorator.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,8 +58,6 @@ def _validate_keyword_arguments(fn_name: str,
# If we have any keyword arguments with a default value that we did not receive an explicit value for, we need
# to bind it, otherwise it will trigger an error when MRC.
for param in param_iter:
if param.annotation is mrc.Subscriber:
continue

if param.default is not signature.empty and param.name not in kwargs:
kwargs[param.name] = param.default
Expand Down Expand Up @@ -226,7 +224,7 @@ def compute_schema_fn_inner(schema: _pipeline.StageSchema):

compute_schema_fn = compute_schema_fn_inner

_validate_keyword_arguments(name, signature, kwargs, param_iter=iter(signature.parameters.values()))
_validate_keyword_arguments(name, signature, kwargs, param_iter=param_iter)

bound_gen_fn = functools.partial(gen_fn, **kwargs)

Expand Down

0 comments on commit 1c82682

Please sign in to comment.