Skip to content

Commit 93c30be

Browse files
add machine env variables
1 parent 60d3424 commit 93c30be

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

src/feelpp/benchmarking/reframe/config/configMachines.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ class MachineConfig(BaseModel):
2929
reports_base_dir:str
3030
input_dataset_base_dir:Optional[str] = None
3131
output_app_dir:str
32+
env_variables:Optional[Dict] = {}
3233
containers:Optional[Dict[str,Container]] = {}
3334

3435
platform:Optional[Literal["apptainer","docker","builtin"]] = "builtin"

src/feelpp/benchmarking/reframe/setup.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,10 @@ def updateConfig(self, replace = None):
4141
"""
4242
self.reader.updateConfig(replace)
4343

44+
def setEnvVariables(self):
45+
for env_var_name,env_var_value in self.reader.config.env_variables.items():
46+
os.environ[env_var_name] = env_var_value
47+
4448

4549
class MachineSetup(Setup):
4650
""" Machine related setup"""
@@ -58,6 +62,7 @@ def setupAfterInit(self,rfm_test,app_config):
5862
Args:
5963
rfm_test (reframe class) : The test to apply the setup
6064
"""
65+
self.setEnvVariables()
6166
self.setValidEnvironments(rfm_test)
6267
self.setTags(rfm_test)
6368
self.setPlatform(rfm_test,app_config)
@@ -137,10 +142,6 @@ def setupBeforeRun(self,rfm_test,machine_config):
137142
def setupAfterInit(self, rfm_test):
138143
self.setEnvVariables()
139144

140-
def setEnvVariables(self):
141-
for env_var_name,env_var_value in self.reader.config.env_variables.items():
142-
os.environ[env_var_name] = env_var_value
143-
144145
def cleanupDirectories(self):
145146
if os.path.exists(self.reader.config.scalability.directory):
146147
shutil.rmtree(self.reader.config.scalability.directory)

0 commit comments

Comments
 (0)