set TMPDIR from script

ITMO-NSS-team · Nov 7, 2023 · 8de268c · 8de268c
1 parent b6d57f5
commit 8de268c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/experiments/fedot_warm_start/config.yaml b/experiments/fedot_warm_start/config.yaml
@@ -1,5 +1,6 @@
 ---
 seed: 42
+tmpdir: '/var/essdata/tmp'
 #data_settings:
 n_datasets: null # null for all available datasets
 test_size: 0.25

diff --git a/experiments/fedot_warm_start/run.py b/experiments/fedot_warm_start/run.py
@@ -3,6 +3,7 @@
 import functools
 import json
 import logging
+import os
 import pickle
 import timeit
 from dataclasses import dataclass, field
@@ -56,6 +57,8 @@
 COLLECT_METRICS = config['collect_metrics']
 COMMON_FEDOT_PARAMS = config['common_fedot_params']
 BASELINE_MODEL = config['baseline_model']
+TMPDIR = config.get('tmpdir')
+
 UPDATE_TRAIN_TEST_DATASETS_SPLIT = config.get('update_train_test_datasets_split')
 
 # Postprocess constants
@@ -324,6 +327,7 @@ def main():
     experiment_date, experiment_date_iso, experiment_date_for_path = get_current_formatted_date()
     save_dir = get_save_dir(experiment_date_for_path)
     setup_logging(save_dir)
+    os.environ.putenv('TMPDIR', TMPDIR)
     progress_file_path = save_dir.joinpath('progress.txt')
     meta_learner_path = save_dir.joinpath('meta_learner.pkl')