lib/jets/lambda/dsl/warnings.rb in jets-5.0.9 vs lib/jets/lambda/dsl/warnings.rb in jets-5.0.10

@@ -21,10 +21,10 @@
     end
 
     def class_iam_policy_unused_warning(managed=false)
       return unless Jets.config.cfn.build.controllers == "one_lambda_for_all_controllers"
       return if self.to_s == "ApplicationController" # ApplicationController not defined in job mode
-      return if self == Jets::PreheatJob
+      return if self.ancestors.include?(Jets::Job::Base)
 
       managed_prefix = managed ? "managed_" : ""
       puts <<~EOL.color(:yellow)
         WARNING: class_#{managed_prefix}iam_policy is not respected when
         config.cfn.build.controllers is not set to "one_lambda_per_controller" or "one_lambda_per_method"