Skip to content

Commit 516df28

Browse files
author
Naresh Kumar Kolloju
committed
change: Rename data file test_hvd_basic.py to hvd_basic.py
With out this, one cannot run non-horvod tests independently with out installing horovod. command "python-k $test_names" is trying to import "test_hvd_basic.py" file as its in data directory
1 parent 11b5442 commit 516df28

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed
File renamed without changes.

tests/integ/test_horovod.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def instance_type(request):
4141
@pytest.mark.canary_quick
4242
def test_horovod(sagemaker_session, instance_type, tmpdir):
4343
job_name = sagemaker.utils.unique_name_from_base('tf-horovod')
44-
estimator = TensorFlow(entry_point=os.path.join(horovod_dir, 'test_hvd_basic.py'),
44+
estimator = TensorFlow(entry_point=os.path.join(horovod_dir, 'hvd_basic.py'),
4545
role='SageMakerRole',
4646
train_instance_count=2,
4747
train_instance_type=instance_type,
@@ -69,7 +69,7 @@ def test_horovod(sagemaker_session, instance_type, tmpdir):
6969
def test_horovod_local_mode(sagemaker_local_session, instances, processes, tmpdir):
7070
output_path = 'file://%s' % tmpdir
7171
job_name = sagemaker.utils.unique_name_from_base('tf-horovod')
72-
estimator = TensorFlow(entry_point=os.path.join(horovod_dir, 'test_hvd_basic.py'),
72+
estimator = TensorFlow(entry_point=os.path.join(horovod_dir, 'hvd_basic.py'),
7373
role='SageMakerRole',
7474
train_instance_count=2,
7575
train_instance_type='local',

0 commit comments

Comments
 (0)