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:
9595 Returns:
9696 str: Path to directory containing the driver script
9797 """
98- pass
9998
10099 @property
101100 @abstractmethod
@@ -108,7 +107,6 @@ def driver_script(self) -> str:
108107 Returns:
109108 str: Name of the driver script file
110109 """
111- pass
112110
113111
114112class Torchrun (DistributedConfig ):
Original file line number Diff line number Diff line change 1111# ANY KIND, either express or implied. See the License for the specific
1212# language governing permissions and limitations under the License.
1313"""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 1111# ANY KIND, either express or implied. See the License for the specific
1212# language governing permissions and limitations under the License.
1313"""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 1111# ANY KIND, either express or implied. See the License for the specific
1212# language governing permissions and limitations under the License.
1313"""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 2323
2424sys .path .insert (0 , str (Path (__file__ ).parent .parent ))
2525
26- from common .utils import ( # noqa: E402
26+ from common .utils import ( # noqa: E402 # pylint: disable=C0413,E0611
2727 logger ,
2828 get_python_executable ,
2929 execute_commands ,
Original file line number Diff line number Diff line change 2929
3030
3131sys .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
3333 logger ,
3434 hyperparameters_to_cli_args ,
3535 get_process_count ,
Original file line number Diff line number Diff line change 1717import sys
1818import subprocess
1919import time
20- import paramiko
2120
2221from pathlib import Path
2322from typing import List
2423
24+ import paramiko
25+
2526sys .path .insert (0 , str (Path (__file__ ).parent .parent ))
2627
27- from common .utils import ( # noqa: E402
28+ from common .utils import ( # noqa: E402 # pylint: disable=C0413,E0611
2829 SM_EFA_NCCL_INSTANCES ,
2930 SM_EFA_RDMA_INSTANCES ,
3031 get_python_executable ,
Original file line number Diff line number Diff line change 2222
2323sys .path .insert (0 , str (Path (__file__ ).parent .parent ))
2424
25- from common .utils import ( # noqa: E402
25+ from common .utils import ( # noqa: E402 # pylint: disable=C0413,E0611
2626 logger ,
2727 hyperparameters_to_cli_args ,
2828 get_process_count ,
Original file line number Diff line number Diff line change 2424
2525sys .path .insert (0 , str (Path (__file__ ).parent .parent ))
2626
27- from common .utils import ( # noqa: E402
27+ from common .utils import ( # noqa: E402 # pylint: disable=C0413,E0611
2828 safe_serialize ,
2929 safe_deserialize ,
3030 read_distributed_json ,
You can’t perform that action at this time.
0 commit comments