"""Placeholder docstring""" from __future__ import absolute_import import importlib import os import shlex import subprocess # pylint: disable=unused-import import sys import textwrap import warnings import six from sagemaker_containers import _env, _errors, _files, _logging, _process logger = _logging.get_logger() DEFAULT_MODULE_NAME = "default_user_module_name" def exists(name): # type: (str) -> bool """Return True if the module exists. Return False otherwise. Args: name (str): module name. Returns: (bool): boolean indicating if the module exists or not. """ try: importlib.import_module(name)
import textwrap from contextlib import contextmanager import signal import os import subprocess import time import sagemaker_containers from retrying import retry from sagemaker_containers import _logging from sagemaker_containers.beta import framework logger = _logging.get_logger() # MPI files. _MPI_SCRIPT = "/mpi_script.sh" _MPI_IS_RUNNING = "/mpi_is_running" _MPI_IS_FINISHED = "/mpi_is_finished" _CHANGE_HOSTNAME_LIBRARY = "/libchangehostname.so" def _change_hostname(current_host): """Compiles a shared library to correct the behavior of the gethostname system call, which OpenMPI depends on. Args: current_host (str): name of the current host, such as algo-1, algo-2, etc. """