diff --git a/examples/pytorch/cpu-demo/demo.py b/examples/pytorch/cpu-demo/demo.py index 8ebb04a82d..3d2d5dc83f 100644 --- a/examples/pytorch/cpu-demo/demo.py +++ b/examples/pytorch/cpu-demo/demo.py @@ -1,4 +1,5 @@ import torch + torch.distributed.init_process_group(init_method="env://") rank = torch.distributed.get_rank() world_size = torch.distributed.get_world_size() diff --git a/examples/pytorch/elastic/echo/echo.py b/examples/pytorch/elastic/echo/echo.py index 17f7632620..67fcf40f72 100644 --- a/examples/pytorch/elastic/echo/echo.py +++ b/examples/pytorch/elastic/echo/echo.py @@ -7,7 +7,6 @@ import torch.distributed as dist - if __name__ == "__main__": env_dict = { diff --git a/sdk/python/kubeflow/training/api_client.py b/sdk/python/kubeflow/training/api_client.py index 52d67804b0..7817ffacbd 100644 --- a/sdk/python/kubeflow/training/api_client.py +++ b/sdk/python/kubeflow/training/api_client.py @@ -12,22 +12,21 @@ import atexit import datetime -from dateutil.parser import parse import json import mimetypes -from multiprocessing.pool import ThreadPool import os import re import tempfile +from multiprocessing.pool import ThreadPool +import kubeflow.training.models # python 2 and python 3 compatibility library import six -from six.moves.urllib.parse import quote - -from kubeflow.training.configuration import Configuration -import kubeflow.training.models +from dateutil.parser import parse from kubeflow.training import rest -from kubeflow.training.exceptions import ApiValueError, ApiException +from kubeflow.training.configuration import Configuration +from kubeflow.training.exceptions import ApiException, ApiValueError +from six.moves.urllib.parse import quote class ApiClient(object):