File tree Expand file tree Collapse file tree 9 files changed +10
-11
lines changed Expand file tree Collapse file tree 9 files changed +10
-11
lines changed Original file line number Diff line number Diff line change @@ -95,7 +95,6 @@ def driver_dir(self) -> str:
95
95
Returns:
96
96
str: Path to directory containing the driver script
97
97
"""
98
- pass
99
98
100
99
@property
101
100
@abstractmethod
@@ -108,7 +107,6 @@ def driver_script(self) -> str:
108
107
Returns:
109
108
str: Name of the driver script file
110
109
"""
111
- pass
112
110
113
111
114
112
class Torchrun (DistributedConfig ):
Original file line number Diff line number Diff line change 11
11
# ANY KIND, either express or implied. See the License for the specific
12
12
# language governing permissions and limitations under the License.
13
13
"""Sagemaker modules container drivers directory."""
14
- from __future__ import absolute_import
14
+ from __future__ import absolute_import
Original file line number Diff line number Diff line change 11
11
# ANY KIND, either express or implied. See the License for the specific
12
12
# language governing permissions and limitations under the License.
13
13
"""Sagemaker modules container drivers - common directory."""
14
- from __future__ import absolute_import
14
+ from __future__ import absolute_import
Original file line number Diff line number Diff line change 11
11
# ANY KIND, either express or implied. See the License for the specific
12
12
# language governing permissions and limitations under the License.
13
13
"""Sagemaker modules container drivers - drivers directory."""
14
- from __future__ import absolute_import
14
+ from __future__ import absolute_import
Original file line number Diff line number Diff line change 23
23
24
24
sys .path .insert (0 , str (Path (__file__ ).parent .parent ))
25
25
26
- from common .utils import ( # noqa: E402
26
+ from common .utils import ( # noqa: E402 # pylint: disable=C0413,E0611
27
27
logger ,
28
28
get_python_executable ,
29
29
execute_commands ,
Original file line number Diff line number Diff line change 29
29
30
30
31
31
sys .path .insert (0 , str (Path (__file__ ).parent .parent ))
32
- from common .utils import ( # noqa: E402
32
+ from common .utils import ( # noqa: E402 # pylint: disable=C0413,E0611
33
33
logger ,
34
34
hyperparameters_to_cli_args ,
35
35
get_process_count ,
Original file line number Diff line number Diff line change 17
17
import sys
18
18
import subprocess
19
19
import time
20
- import paramiko
21
20
22
21
from pathlib import Path
23
22
from typing import List
24
23
24
+ import paramiko
25
+
25
26
sys .path .insert (0 , str (Path (__file__ ).parent .parent ))
26
27
27
- from common .utils import ( # noqa: E402
28
+ from common .utils import ( # noqa: E402 # pylint: disable=C0413,E0611
28
29
SM_EFA_NCCL_INSTANCES ,
29
30
SM_EFA_RDMA_INSTANCES ,
30
31
get_python_executable ,
Original file line number Diff line number Diff line change 22
22
23
23
sys .path .insert (0 , str (Path (__file__ ).parent .parent ))
24
24
25
- from common .utils import ( # noqa: E402
25
+ from common .utils import ( # noqa: E402 # pylint: disable=C0413,E0611
26
26
logger ,
27
27
hyperparameters_to_cli_args ,
28
28
get_process_count ,
Original file line number Diff line number Diff line change 24
24
25
25
sys .path .insert (0 , str (Path (__file__ ).parent .parent ))
26
26
27
- from common .utils import ( # noqa: E402
27
+ from common .utils import ( # noqa: E402 # pylint: disable=C0413,E0611
28
28
safe_serialize ,
29
29
safe_deserialize ,
30
30
read_distributed_json ,
You can’t perform that action at this time.
0 commit comments