@@ -560,10 +560,11 @@ def test_find_project_root_not_found(self):
560560 @pytest .mark .parametrize (
561561 "version,expected" ,
562562 [
563+ ("3.12" , "312" ),
563564 ("3.11" , "311" ),
564565 ("3.10" , "310" ),
565566 ("3.9" , "39" ),
566- ("" , "311 " ), # Default
567+ ("" , "312 " ), # Default
567568 ],
568569 )
569570 def test_validate_python_version_valid (self , version , expected ):
@@ -619,7 +620,7 @@ def test_prepare_build_environment_cpu(self):
619620 build_params = {
620621 "gpu" : False ,
621622 "llm_runtime" : "transformers" ,
622- "python_version" : "3.11 " ,
623+ "python_version" : "3.12 " ,
623624 "system_packages" : ["git" , "curl" ],
624625 "python_packages" : ["numpy" , "pandas" ],
625626 }
@@ -640,7 +641,7 @@ def test_prepare_build_environment_cpu(self):
640641
641642 assert llm_runtime == "transformers"
642643 assert ray_version == "2.47.0"
643- assert python_version == "-py311 "
644+ assert python_version == "-py312 "
644645 assert cuda_version == ""
645646 assert device_type == "-cpu"
646647 assert system_pkg_str == "git curl"
@@ -655,7 +656,7 @@ def test_prepare_build_environment_gpu(self):
655656 "gpu" : True ,
656657 "cuda_version" : "12.8" ,
657658 "llm_runtime" : "vllm" ,
658- "python_version" : "3.11 " ,
659+ "python_version" : "3.12 " ,
659660 }
660661
661662 result = prepare_build_environment (build_params )
@@ -674,7 +675,7 @@ def test_prepare_build_environment_gpu(self):
674675
675676 assert llm_runtime == "vllm"
676677 assert ray_version == "2.47.0"
677- assert python_version == "-py311 "
678+ assert python_version == "-py312 "
678679 assert cuda_version == "-cu128"
679680 assert device_type == "-gpu"
680681 assert system_pkg_str == ""
@@ -721,7 +722,7 @@ def test_prepare_build_command(self):
721722 False , # no_cache
722723 "transformers" , # llm_runtime
723724 "2.47.0" , # ray_version
724- "-py311 " , # python_version
725+ "-py312 " , # python_version
725726 "" , # cuda_version
726727 "-cpu" , # device_type
727728 "numpy pandas" , # python_pkg_str
0 commit comments