From 4f8a9aca405ea0b4918c129ae02e649c55ea79c5 Mon Sep 17 00:00:00 2001
From: XianBW <36835909+XianBW@users.noreply.github.com>
Date: Sun, 29 Sep 2024 11:29:40 +0800
Subject: [PATCH] convert covid workspace to merged.py for submission (#384)

---
 rdagent/app/kaggle/loop.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/rdagent/app/kaggle/loop.py b/rdagent/app/kaggle/loop.py
index 9642e062..1cd8de5e 100644
--- a/rdagent/app/kaggle/loop.py
+++ b/rdagent/app/kaggle/loop.py
@@ -89,7 +89,10 @@ class KaggleRDLoop(RDLoop):
                 exp = self.model_runner.develop(prev_out["coding"])
             logger.log_object(exp, tag="runner result")
 
-            if KAGGLE_IMPLEMENT_SETTING.competition in ["optiver-realized-volatility-prediction"]:
+            if KAGGLE_IMPLEMENT_SETTING.competition in [
+                "optiver-realized-volatility-prediction",
+                "covid19-global-forecasting-week-1",
+            ]:
                 try:
                     python_files_to_notebook(
                         KAGGLE_IMPLEMENT_SETTING.competition, exp.experiment_workspace.workspace_path