[Bug] Installing Pyxis if nvidia is installed or enabled

cookbooks/aws-parallelcluster-slurm/recipes/install/install_pyxis.rb

@@ -15,7 +15,7 @@
 # OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions and
 # limitations under the License.
 
-return unless nvidia_installed?
+return unless nvidia_enabled? || nvidia_installed?
 return if pyxis_installed?
 
 pyxis_version = node['cluster']['pyxis']['version']

cookbooks/aws-parallelcluster-slurm/spec/unit/recipes/install_pyxis_spec.rb

@@ -33,6 +33,7 @@
           node.override['cluster']['pyxis']['version'] = pyxis_version
           node.override['cluster']['pyxis']['runtime_path'] = pyxis_runtime_dir
         end
+        allow_any_instance_of(Object).to receive(:nvidia_enabled?).and_return(true)
         allow_any_instance_of(Object).to receive(:nvidia_installed?).and_return(true)
         allow_any_instance_of(Object).to receive(:pyxis_installed?).and_return(false)
         runner.converge(described_recipe)
@@ -93,6 +94,7 @@
           runner = runner(platform: platform, version: version) do |_node|
             RSpec::Mocks.configuration.allow_message_expectations_on_nil = true
           end
+          allow_any_instance_of(Object).to receive(:nvidia_enabled?).and_return(true)
           allow_any_instance_of(Object).to receive(:nvidia_installed?).and_return(true)
           allow_any_instance_of(Object).to receive(:pyxis_installed?).and_return(true)
           runner.converge(described_recipe)

cookbooks/aws-parallelcluster-slurm/test/controls/pyxis_spec.rb

@@ -9,25 +9,24 @@
 # This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied.
 # See the License for the specific language governing permissions and limitations under the License.
 
-# FIXME: Re-enabled the following check and fix failures
-# control 'tag:install_pyxis_installed' do
-#   only_if { instance.nvidia_installed? }
-#
-#   title 'Checks Pyxis has been installed'
-#
-#   examples_dir = "/opt/parallelcluster/examples"
-#   dirs = [ examples_dir, "#{examples_dir}/spank", "#{examples_dir}/pyxis" ]
-#   dirs.each do |path|
-#     describe directory(path) do
-#       it { should exist }
-#     end
-#   end
-#
-#   describe file("#{examples_dir}/pyxis/pyxis.conf") do
-#     it { should exist }
-#   end
-#
-#   describe file("#{examples_dir}/spank/plugstack.conf") do
-#     it { should exist }
-#   end
-# end
+control 'tag:install_pyxis_installed' do
+  only_if { ['yes', true, 'true'].include?(node['cluster']['nvidia']['enabled']) || instance.nvidia_installed? }
+
+  title 'Checks Pyxis has been installed'
+
+  examples_dir = "/opt/parallelcluster/examples"
+  dirs = [ examples_dir, "#{examples_dir}/spank", "#{examples_dir}/pyxis" ]
+  dirs.each do |path|
+    describe directory(path) do
+      it { should exist }
+    end
+  end
+
+  describe file("#{examples_dir}/pyxis/pyxis.conf") do
+    it { should exist }
+  end
+
+  describe file("#{examples_dir}/spank/plugstack.conf") do
+    it { should exist }
+  end
+end