diff --git a/rudra/deployments/emr_scripts/npm_emr.py b/rudra/deployments/emr_scripts/npm_emr.py index 8309491..ea617f1 100644 --- a/rudra/deployments/emr_scripts/npm_emr.py +++ b/rudra/deployments/emr_scripts/npm_emr.py @@ -27,7 +27,7 @@ def run_job(self, input_dict): emr_config_obj = EMRConfig(name=name, s3_bootstrap_uri=bootstrap_uri, training_repo_url=self.training_repo_url, - training_file_name='recommendation_engine/autoencoder/train/train.py' + training_file_name='recommendation_engine/autoencoder/train/train.py', instance_type='p3.2xlarge', log_uri=log_uri, ecosystem=self.ecosystem,