Skip to content

Commit 5a89ce2

Browse files
committed
Update ChainDeployment.load_chain() to take kwargs.
1 parent 534fe95 commit 5a89ce2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ads/llm/deploy.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,5 +55,5 @@ def prepare(self, **kwargs) -> GenericModel:
5555
os.unlink(score_py.name)
5656

5757
@classmethod
58-
def load_chain(cls, yaml_uri: str) -> Any:
59-
return load_from_yaml(yaml_uri)
58+
def load_chain(cls, yaml_uri: str, **kwargs) -> Any:
59+
return load_from_yaml(yaml_uri, **kwargs)

0 commit comments

Comments
 (0)