Skip to content

Commit 287c78d

Browse files
committed
fix: some integ tests missing required args
1 parent 8fd00fc commit 287c78d

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

tests/integ/test_chainer_train.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,14 +100,16 @@ def test_attach_deploy(sagemaker_session, chainer_full_version, cpu_instance_typ
100100

101101

102102
@pytest.mark.local_mode
103-
def test_deploy_model(chainer_local_training_job, sagemaker_local_session):
103+
def test_deploy_model(chainer_local_training_job, sagemaker_local_session, chainer_full_version):
104104
script_path = os.path.join(DATA_DIR, "chainer_mnist", "mnist.py")
105105

106106
model = ChainerModel(
107107
chainer_local_training_job.model_data,
108108
"SageMakerRole",
109109
entry_point=script_path,
110110
sagemaker_session=sagemaker_local_session,
111+
framework_version=chainer_full_version,
112+
py_version=PYTHON_VERSION,
111113
)
112114

113115
predictor = model.deploy(1, "local")

tests/integ/test_tuner.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -689,14 +689,15 @@ def test_tuning_tf_vpc_multi(sagemaker_session, cpu_instance_type):
689689

690690

691691
@pytest.mark.canary_quick
692-
def test_tuning_chainer(sagemaker_session, cpu_instance_type):
692+
def test_tuning_chainer(sagemaker_session, chainer_full_version, cpu_instance_type):
693693
with timeout(minutes=TUNING_DEFAULT_TIMEOUT_MINUTES):
694694
script_path = os.path.join(DATA_DIR, "chainer_mnist", "mnist.py")
695695
data_path = os.path.join(DATA_DIR, "chainer_mnist")
696696

697697
estimator = Chainer(
698698
entry_point=script_path,
699699
role="SageMakerRole",
700+
framework_version=chainer_full_version,
700701
py_version=PYTHON_VERSION,
701702
train_instance_count=1,
702703
train_instance_type=cpu_instance_type,

0 commit comments

Comments
 (0)