Path	Lines of Code
test/__init__.py	188
test/conftest.py	52
test/container/dummy/train.py	1
test/container/tensorflow/train.py	2
test/fake_ml_framework.py	26
test/functional/__init__.py	1
test/functional/simple_framework.py	13
test/functional/test_download_and_import.py	105
test/functional/test_intermediate_output.py	187
test/functional/test_mpi.py	40
test/functional/test_training_framework.py	282
test/integration/local/test_dummy.py	28
test/integration/local/test_tensorflow.py	27
test/resources/openmpi/launcher.sh	2
test/resources/openmpi/script.py	15
test/unit/__init__.py	1
test/unit/c/test_gethostname.py	41
test/unit/cli/__init__.py	1
test/unit/cli/test_train.py	6
test/unit/dummy/__init__.py	1
test/unit/dummy/dummy.py	2
test/unit/dummy/tensorflow/__init__.py	1
test/unit/dummy/tensorflow/compiler/__init__.py	1
test/unit/dummy/tensorflow/compiler/xla/__init__.py	1
test/unit/dummy/tensorflow/compiler/xla/dummy_xla.py	2
test/unit/test_encoder.py	158
test/unit/test_entry_point.py	171
test/unit/test_entry_point_type.py	22
test/unit/test_environment.py	243
test/unit/test_errors.py	34
test/unit/test_files.py	109
test/unit/test_functions.py	35
test/unit/test_intermediate_output.py	100
test/unit/test_mapping.py	183
test/unit/test_modules.py	244
test/unit/test_mpi.py	538
test/unit/test_process.py	222
test/unit/test_pytorch_xla.py	219
test/unit/test_runner.py	110
test/unit/test_smdataparallel.py	374
test/unit/test_timeout.py	10
test/unit/test_torch_distributed.py	356
test/unit/test_trainer.py	163