lib/elasticity/job_flow.rb in elasticity-6.0.8 vs lib/elasticity/job_flow.rb in elasticity-6.0.9

@@ -131,9 +131,21 @@
     def set_task_instance_group(instance_group)
       instance_group.role = 'TASK'
       @instance_groups[:task] = instance_group
     end
 
+    def set_master_ebs_configuration(ebs_configuration)
+      @instance_groups[:master].set_ebs_configuration(ebs_configuration)
+    end
+
+    def set_core_ebs_configuration(ebs_configuration)
+      @instance_groups[:core].set_ebs_configuration(ebs_configuration)
+    end
+
+    def set_task_ebs_configuration(ebs_configuration)
+      @instance_groups[:task].set_ebs_configuration(ebs_configuration)
+    end
+
     def add_step(jobflow_step)
       if is_jobflow_running?
         jobflow_steps = []
         if jobflow_step.requires_installation? && !@installed_steps.include?(jobflow_step.class)
           jobflow_steps.concat(jobflow_step.aws_installation_steps)