bolt-modules/boltlib/lib/puppet/functions/set_feature.rb in bolt-1.31.1 vs bolt-modules/boltlib/lib/puppet/functions/set_feature.rb in bolt-1.32.0

@@ -20,9 +20,10 @@
   #   set_feature($target, 'puppet-agent', true)
   dispatch :set_feature do
     param 'Target', :target
     param 'String', :feature
     optional_param 'Boolean', :value
+    return_type 'Target'
   end
 
   def set_feature(target, feature, value = true)
     unless Puppet[:tasks]
       raise Puppet::ParseErrorWithIssue