diff --git a/pdelfin/beakerpipeline.py b/pdelfin/beakerpipeline.py index f3ad5a5..bef0ab8 100644 --- a/pdelfin/beakerpipeline.py +++ b/pdelfin/beakerpipeline.py @@ -688,7 +688,7 @@ def submit_beaker_job(args): EnvVar(name="AWS_CREDENTIALS_FILE", secret=f"{owner}-AWS_CREDENTIALS_FILE"), ], resources=TaskResources(gpu_count=1), - constraints=Constraints(cluster=args.beaker_cluster), + constraints=Constraints(cluster=args.beaker_cluster if isinstance(args.beaker_cluster, list) else [args.beaker_cluster]), result=ResultSpec(path="/noop-results"), ) ], diff --git a/pyproject.toml b/pyproject.toml index 5338d7a..5a81c16 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -73,6 +73,7 @@ dev = [ inference = [ #"sglang["all"] #"git+https://github.com/sgl-project/sglang.git@eff468dd5a3d24646560eb044276585f7a11ac3c#subdirectory=python&egg=sglang[all]" + "beaker-py", ] train = [