Skip to content

Commit 43aabc5

Browse files
authored
fix: post training algorithm param (#310)
was getting: ``` ╰─ llama-stack-client --version Traceback (most recent call last): File "/Users/charliedoern/projects/Documents/llama-stack/venv/bin/llama-stack-client", line 4, in <module> from llama_stack_client.lib.cli.llama_stack_client import main File "/Users/charliedoern/projects/Documents/llama-stack-client-python/src/llama_stack_client/lib/cli/llama_stack_client.py", line 23, in <module> from .post_training import post_training File "/Users/charliedoern/projects/Documents/llama-stack-client-python/src/llama_stack_client/lib/cli/post_training/__init__.py", line 7, in <module> from .post_training import post_training File "/Users/charliedoern/projects/Documents/llama-stack-client-python/src/llama_stack_client/lib/cli/post_training/post_training.py", line 12, in <module> from llama_stack_client.types.alpha.post_training_supervised_fine_tune_params import ( ImportError: cannot import name 'AlgorithmConfigParam' from 'llama_stack_client.types.alpha.post_training_supervised_fine_tune_params' (/Users/charliedoern/projects/Documents/llama-stack-client-python/src/llama_stack_client/types/alpha/post_training_supervised_fine_tune_params.py). Did you mean: 'AlgorithmConfig'? ``` fixes that. Signed-off-by: Charlie Doern <cdoern@redhat.com>
1 parent 15fa286 commit 43aabc5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/llama_stack_client/lib/cli/post_training/post_training.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from rich.console import Console
1111

1212
from llama_stack_client.types.alpha.post_training_supervised_fine_tune_params import (
13-
AlgorithmConfigParam,
13+
AlgorithmConfig,
1414
TrainingConfig,
1515
)
1616

@@ -36,7 +36,7 @@ def supervised_fine_tune(
3636
ctx,
3737
job_uuid: str,
3838
model: str,
39-
algorithm_config: AlgorithmConfigParam,
39+
algorithm_config: AlgorithmConfig,
4040
training_config: TrainingConfig,
4141
checkpoint_dir: Optional[str],
4242
):

0 commit comments

Comments
 (0)