[Dependencies] Overriding the NVIDIA version for Centos7 using node_attributes (#2657)

* [Dependencies] Overriding the NVIDIA version for Centos7 using node_attributes

* Revert "[kitchen tests] Fix pinning NVIDIA version for Centos7 to 535.129.03 (#2648)"

This reverts commit 295f44c.

---------

Co-authored-by: Himani Deshpande <[email protected]>

Author: himani2411

cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/partial/_nvidia_driver_common.rb

@@ -23,6 +23,10 @@
   return unless nvidia_driver_enabled?
   return if on_docker?
 
+  # Share nvidia driver version with InSpec tests
+  node.default['cluster']['nvidia']['driver_version'] = _nvidia_driver_version
+  node_attributes "Save Nvidia driver version for Inspec tests"
+
   remote_file tmp_nvidia_run do
     source nvidia_driver_url
     mode '0755'

cookbooks/aws-parallelcluster-platform/spec/unit/resources/nvidia_driver_spec.rb

@@ -187,6 +187,10 @@ def self.setup(chef_run, nvidia_driver_version: nil)
         end
         cached(:node) { chef_run.node }
 
+        it 'dumps nodes attribues' do
+          is_expected.to write_node_attributes('Save Nvidia driver version for Inspec tests')
+        end
+
         it 'sets up nvidia_driver' do
           is_expected.to setup_nvidia_driver('setup')
         end

cookbooks/aws-parallelcluster-platform/test/controls/nvidia_spec.rb

@@ -15,11 +15,7 @@
       (node['cluster']['nvidia']['enabled'] == 'yes' || node['cluster']['nvidia']['enabled'] == true)
   end
 
-  expected_nvidia_driver_version = if os_properties.centos7?
-                                     '535.129.03'
-                                   else
-                                     node['cluster']['nvidia']['driver_version']
-                                   end
+  expected_nvidia_driver_version = node['cluster']['nvidia']['driver_version']
   expected_nvidia_kernel_license = 'Dual MIT/GPL'
   expected_nvidia_kernel_module = "NVRM version: NVIDIA UNIX Open Kernel Module"