[local support] set related environ var for local running (AFLplusplu…

…s#441) Related to google/fuzzbench#251
mattweingarten · Jun 16, 2020 · 8ce3eb6 · 8ce3eb6
1 parent 80e89d1
commit 8ce3eb6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/experiment/run_experiment.py b/experiment/run_experiment.py
@@ -287,8 +287,11 @@ def filter_file(tar_info):
             return None
         return tar_info
 
-    experiment_filestore_path = os.path.join(config['experiment_filestore'],
-                                             config['experiment'])
+    # Set environment variables to use corresponding filestore_utils.
+    os.environ['EXPERIMENT_FILESTORE'] = config['experiment_filestore']
+    os.environ['EXPERIMENT'] = config['experiment']
+    experiment_filestore_path = experiment_utils.get_experiment_filestore_path()
+
     base_destination = os.path.join(experiment_filestore_path, 'input')
 
     # Send the local source repository to the cloud for use by dispatcher.