v2.15.0

Author: jbernal0019

README.rst

@@ -18,7 +18,7 @@ relevant modules.
 Overview
 --------
 
-At time of writing (early 2022), three scripts/modules are in production:
+Three scripts/modules are included in this repository:
 
 - a general CUBE client
 - a more specific plugin search utility

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "python-chrisclient"
-version = "2.14.0"
+version = "2.15.0"
 authors = [
   { name = "FNNDSC", email = "dev@babymri.org" },
 ]

src/chrisclient/client.py

@@ -420,6 +420,8 @@ def compute_workflow_nodes_info(self, pipeline_default_parameters,
                     'title': default_param['plugin_piping_title'],
                     'cpu_limit': default_param['plugin_piping_cpu_limit'],
                     'memory_limit': default_param['plugin_piping_memory_limit'],
+                    'gpu_limit': default_param['plugin_piping_gpu_limit'],
+                    'number_of_workers': default_param['plugin_piping_number_of_workers'],
                     'plugin_parameter_defaults': []
                 }
 

tests/test_client.py

@@ -121,6 +121,8 @@ def test_get_pipeline_default_parameters(self):
         self.assertIn('plugin_piping_title', first_param)
         self.assertIn('plugin_piping_cpu_limit', first_param)
         self.assertIn('plugin_piping_memory_limit', first_param)
+        self.assertIn('plugin_piping_gpu_limit', first_param)
+        self.assertIn('plugin_piping_number_of_workers', first_param)
 
     def test_get_pipeline_default_parameters_unauthenticated(self):
         """
@@ -180,6 +182,8 @@ def test_compute_workflow_nodes_info(self):
             self.assertIn('title', node)
             self.assertIn('cpu_limit', node)
             self.assertIn('memory_limit', node)
+            self.assertIn('gpu_limit', node)
+            self.assertIn('number_of_workers', node)
             if 'plugin_parameter_defaults' in node:
                 for param in node['plugin_parameter_defaults']:
                     self.assertIsNone(param['default'])