Skip to content

Commit 83f9e2c

Browse files
authored
Move processing of requirements file in. (#35)
Fix test to not rely on local copy of the code.
1 parent 6a3c770 commit 83f9e2c

File tree

2 files changed

+1
-1
lines changed

2 files changed

+1
-1
lines changed

src/tf_container/train_entry_point.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,7 @@ def train():
137137
os.environ['S3_REQUEST_TIMEOUT_MSEC'] = str(env.hyperparameters.get('s3_checkpoint_save_timeout', 60000))
138138

139139
env.download_user_module()
140+
env.pip_install_requirements()
140141

141142
customer_script = env.import_user_module()
142143

test/integ/test_estimator_classification.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
def test_estimator_classification(docker_image, sagemaker_session, opt_ml, processor):
2424
resource_path = os.path.join(SCRIPT_PATH, '../resources/iris')
2525

26-
copy_resource(resource_path, opt_ml, 'code')
2726
copy_resource(resource_path, opt_ml, 'data', 'input/data')
2827

2928
s3_source_archive = fw_utils.tar_and_upload_dir(session=sagemaker_session.boto_session,

0 commit comments

Comments
 (0)