lib/buildkite/builder/extension.rb in buildkite-builder-2.0.0.beta1 vs lib/buildkite/builder/extension.rb in buildkite-builder-2.0.0.beta2

@@ -26,10 +26,14 @@
         # Override to provide extra functionality.
       end
 
       private
 
+      def log
+        context.logger
+      end
+
       def buildkite
         @buildkite ||= begin
           unless Buildkite.env
             raise 'Must be in Buildkite environment to access the Buildkite API'
           end
@@ -41,11 +45,10 @@
       def prepare
         # Override to provide extra functionality.
       end
 
       def pipeline(&block)
-        context.dsl.instance_eval(&block) if block_given?
-        context
+        context.dsl.instance_eval(&block)
       end
     end
   end
 end