File tree Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Original file line number Diff line number Diff line change 18
18
"""
19
19
from __future__ import print_function , absolute_import
20
20
21
+ import json
21
22
import os
22
23
import pathlib
23
24
import logging
@@ -1313,7 +1314,7 @@ def __init__(
1313
1314
inter-container traffic, security group IDs, and subnets (default: None).
1314
1315
"""
1315
1316
if not command :
1316
- command = ["python " ]
1317
+ command = ["python3 " ]
1317
1318
1318
1319
self .estimator_cls = estimator_cls
1319
1320
self .framework_version = framework_version
@@ -1644,14 +1645,11 @@ def _generate_framework_script(self, user_script: str) -> str:
1644
1645
1645
1646
subprocess.run(["pip", "install", "-r", "requirements.txt"])
1646
1647
1647
- cmd = [" {entry_point_command}", "{entry_point}"] + sys.argv[1:]
1648
+ cmd = {entry_point_command} + sys.argv[1:]
1648
1649
print(' '.join(cmd))
1649
1650
subprocess.run(cmd)
1650
1651
"""
1651
- ).format (
1652
- entry_point_command = " " .join (self .command ),
1653
- entry_point = user_script ,
1654
- )
1652
+ ).format (entry_point_command = json .dumps (self .command + [user_script ]))
1655
1653
1656
1654
def _upload_payload (
1657
1655
self ,
You can’t perform that action at this time.
0 commit comments