From 364317d79dead0731ce0518fe0054d8c11afb6c3 Mon Sep 17 00:00:00 2001
From: nimishsantosh107 <nimishsantosh107@icloud.com>
Date: Wed, 5 Jan 2022 20:29:46 +0300
Subject: [PATCH] added seed logging in service for actions
---
flatland/evaluators/aicrowd_helpers.py | 2 ++
flatland/evaluators/service.py | 3 +++
2 files changed, 5 insertions(+)
diff --git a/flatland/evaluators/aicrowd_helpers.py b/flatland/evaluators/aicrowd_helpers.py
index 610956f4..2c18deb3 100644
--- a/flatland/evaluators/aicrowd_helpers.py
+++ b/flatland/evaluators/aicrowd_helpers.py
@@ -109,6 +109,8 @@ def upload_folder_to_s3(folderpath):
file_target_key = f'analysis_logs/{get_submission_id()}/{path[path.find(next(filter(str.isalpha, path))):]}/{file}'
localpath = os.path.join(path, file)
+ print(f"[INFO] SAVING: {localpath}")
+
s3.put_object(
ACL=S3_BUCKET_ACL,
Bucket=S3_BUCKET,
diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py
index 4d9b8213..a7dc1590 100644
--- a/flatland/evaluators/service.py
+++ b/flatland/evaluators/service.py
@@ -8,6 +8,7 @@ import shutil
import time
import traceback
import json
+import yaml
import itertools
import re
@@ -132,6 +133,8 @@ class FlatlandRemoteEvaluationService:
self.action_dir = action_dir
if action_dir and not os.path.exists(self.action_dir):
os.makedirs(self.action_dir)
+ with open(os.path.join(self.action_dir, 'seed.yml'), 'w') as outfile:
+ yaml.dump({"RANDOM_SEED": RANDOM_SEED}, outfile, default_flow_style=False)
self.episode_dir = episode_dir
if episode_dir and not os.path.exists(self.episode_dir):
os.makedirs(self.episode_dir)
--
GitLab