Commit 8b84605
committed
fix: apply JumpStart model spec enable_network_isolation in ModelBuilder
In v2, JumpStart models deployed with EnableNetworkIsolation=True because
the JumpStartModel class read inference_enable_network_isolation from the
model spec JSON. In v3's ModelBuilder, this field was never extracted from
the JumpStart init_kwargs or deploy_kwargs, causing all JumpStart models
to deploy without network isolation (defaulting to False).
This fix applies enable_network_isolation from the JumpStart model spec
in both code paths:
1. _build_for_jumpstart() - regular build path via get_init_kwargs()
2. from_jumpstart_config() - classmethod path via _retrieve_model_deploy_kwargs()
The user-provided value (via network= param) takes precedence.1 parent ab5f478 commit 8b84605
2 files changed
Lines changed: 13 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
3621 | 3621 | | |
3622 | 3622 | | |
3623 | 3623 | | |
| 3624 | + | |
| 3625 | + | |
| 3626 | + | |
| 3627 | + | |
| 3628 | + | |
| 3629 | + | |
3624 | 3630 | | |
3625 | 3631 | | |
3626 | 3632 | | |
| |||
Lines changed: 7 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
877 | 877 | | |
878 | 878 | | |
879 | 879 | | |
| 880 | + | |
| 881 | + | |
| 882 | + | |
| 883 | + | |
| 884 | + | |
| 885 | + | |
| 886 | + | |
880 | 887 | | |
881 | 888 | | |
882 | 889 | | |
| |||
0 commit comments