File tree Expand file tree Collapse file tree 4 files changed +861
-3
lines changed
data/tensorflow_mnist/transform Expand file tree Collapse file tree 4 files changed +861
-3
lines changed Original file line number Diff line number Diff line change @@ -590,6 +590,9 @@ def create_model(
590
590
if "name" not in kwargs :
591
591
kwargs ["name" ] = self ._current_job_name
592
592
593
+ if "enable_network_isolation" not in kwargs :
594
+ kwargs ["enable_network_isolation" ] = self .enable_network_isolation ()
595
+
593
596
if endpoint_type == "tensorflow-serving" or self ._script_mode_enabled ():
594
597
return self ._create_tfs_model (
595
598
role = role ,
@@ -630,7 +633,6 @@ def _create_tfs_model(
630
633
entry_point = entry_point ,
631
634
source_dir = source_dir ,
632
635
dependencies = dependencies ,
633
- enable_network_isolation = self .enable_network_isolation (),
634
636
** kwargs
635
637
)
636
638
@@ -660,7 +662,6 @@ def _create_default_model(
660
662
sagemaker_session = self .sagemaker_session ,
661
663
vpc_config = self .get_vpc_config (vpc_config_override ),
662
664
dependencies = dependencies or self .dependencies ,
663
- enable_network_isolation = self .enable_network_isolation (),
664
665
** kwargs
665
666
)
666
667
You can’t perform that action at this time.
0 commit comments