Add kestrel as an exec option

NREL · Nov 15, 2023 · 96c3ad5 · 96c3ad5
1 parent 8ad8c9d
commit 96c3ad5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/reV/config/base_analysis_config.py b/reV/config/base_analysis_config.py
@@ -102,6 +102,7 @@ def execution_control(self):
             ec_config_types = {'local': BaseExecutionConfig,
                                'slurm': SlurmConfig,
                                'eagle': SlurmConfig,
+                               'kestrel': SlurmConfig,
                                }
             if 'option' in ec:
                 try:

diff --git a/reV/config/execution.py b/reV/config/execution.py
@@ -157,13 +157,13 @@ class SlurmConfig(HPCConfig):
 
     def _preflight(self):
         """Run a preflight check on the config."""
-        if self.option == 'eagle':
+        if self.option in {'eagle', 'kestrel', 'slurm'}:
             if self.allocation is None:
-                msg = 'Eagle execution config must have an "allocation" input'
+                msg = 'HPC execution config must have an "allocation" input'
                 logger.error(msg)
                 raise ConfigError(msg)
             if self.walltime is None:
-                msg = 'Eagle execution config must have an "walltime" input'
+                msg = 'HPC execution config must have a "walltime" input'
                 logger.error(msg)
                 raise ConfigError(msg)