Skip to content

Commit

Permalink
input_shape should be input_shapes (#1282)
Browse files Browse the repository at this point in the history
  • Loading branch information
Satrat committed Sep 26, 2023
1 parent 5b33261 commit f89b5ae
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/deepsparse/base_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@ def create(
if issubclass(
pipeline_constructor, Bucketable
) and pipeline_constructor.should_bucket(**kwargs):
if kwargs.get("input_shape", None):
if kwargs.get("input_shapes", None):
raise ValueError(
"Overriding input shapes not supported with Bucketing enabled"
)
Expand Down

0 comments on commit f89b5ae

Please sign in to comment.