more fixes

executorlib/standalone/inputcheck.py

@@ -169,7 +169,7 @@ def check_pysqa_config_directory(pysqa_config_directory: Optional[str]) -> None:
         )
 
 
-def validate_number_of_cores(max_cores: int, max_workers: int) -> int:
+def validate_number_of_cores(max_cores: Optional[int], max_workers: Optional[int]) -> int:
     """
     Validate the number of cores and return the appropriate value.
     """

tests/test_shared_input_check.py

@@ -17,6 +17,7 @@
     check_max_workers_and_cores,
     check_hostname_localhost,
     check_pysqa_config_directory,
+    validate_number_of_cores,
 )
 
 
@@ -80,9 +81,9 @@ def test_check_flux_executor_pmi_mode(self):
 
     def test_check_max_workers_and_cores(self):
         with self.assertRaises(ValueError):
-            check_max_workers_and_cores(max_workers=2, max_cores=1)
+            check_max_workers_and_cores(max_workers=2, max_cores=None)
         with self.assertRaises(ValueError):
-            check_max_workers_and_cores(max_workers=1, max_cores=2)
+            check_max_workers_and_cores(max_workers=None, max_cores=2)
         with self.assertRaises(ValueError):
             check_max_workers_and_cores(max_workers=2, max_cores=2)
 
@@ -95,3 +96,8 @@ def test_check_hostname_localhost(self):
     def test_check_pysqa_config_directory(self):
         with self.assertRaises(ValueError):
             check_pysqa_config_directory(pysqa_config_directory="path/to/config")
+
+    def test_validate_number_of_cores(self):
+        self.assertIsInstance(validate_number_of_cores(max_cores=None, max_workers=None), int)
+        self.assertIsInstance(validate_number_of_cores(max_cores=1, max_workers=None), int)
+        self.assertIsInstance(validate_number_of_cores(max_cores=None, max_workers=1), int)