From 86151bf856af597630e2400e069cb10975565e3e Mon Sep 17 00:00:00 2001
From: lkugler <lukas.kugler@gmail.com>
Date: Fri, 10 Mar 2023 17:20:32 +0100
Subject: [PATCH] missing imports
---
cycled_exp.py | 38 ++++++++++++++++++--------------------
dartwrf/workflows.py | 4 ++--
2 files changed, 20 insertions(+), 22 deletions(-)
diff --git a/cycled_exp.py b/cycled_exp.py
index d130974..1561029 100755
--- a/cycled_exp.py
+++ b/cycled_exp.py
@@ -3,15 +3,13 @@ import os, sys, shutil, glob, warnings
import datetime as dt
from dartwrf.utils import script_to_str
-from config.cfg import exp
-from config.clusters import cluster
-from dartwrf.workflows import *
+from dartwrf.workflows import WorkFlows
if __name__ == "__main__":
"""
Run a cycled OSSE with WRF and DART.
"""
- cluster.setup()
+ w = WorkFlows(exp_config='cfg.py', server_config='jet.py')
timedelta_integrate = dt.timedelta(minutes=15)
timedelta_btw_assim = dt.timedelta(minutes=15)
@@ -26,20 +24,20 @@ if __name__ == "__main__":
last_assim_time = dt.datetime(2008, 7, 30, 13,30)
forecast_until = dt.datetime(2008, 7, 30, 18)
- prepare_WRFrundir(init_time)
- # id = run_ideal(depends_on=id)
- # id = wrfinput_insert_wbubble(depends_on=id)
+ w.prepare_WRFrundir(init_time)
+ # id = w.run_ideal(depends_on=id)
+ # id = w.wrfinput_insert_wbubble(depends_on=id)
if True: # random
prior_path_exp = '/jetfs/home/lkugler/data/sim_archive/exp_v1.19_P2_noDA'
- init_time = dt.datetime(2008, 7, 30, 13)
- time = dt.datetime(2008, 7, 30, 14)
+ init_time = dt.datetime(2008, 7, 30, 12)
+ time = dt.datetime(2008, 7, 30, 13)
last_assim_time = dt.datetime(2008, 7, 30, 14)
- forecast_until = dt.datetime(2008, 7, 30, 14, 15)
+ forecast_until = dt.datetime(2008, 7, 30, 14, 18)
- prepare_WRFrundir(init_time)
- # id = run_ideal(depends_on=id)
+ w.prepare_WRFrundir(init_time)
+ # id = w.run_ideal(depends_on=id)
# prior_path_exp = cluster.archivedir
# prior_path_exp = '/gpfs/data/fs71386/lkugler/sim_archive/exp_v1.19_P5+su_noDA'
@@ -54,13 +52,13 @@ if __name__ == "__main__":
# i.e. 13z as a prior to assimilate 12z observations
prior_valid_time = time
- id = assimilate(time, prior_init_time, prior_valid_time, prior_path_exp, depends_on=id)
+ id = w.assimilate(time, prior_init_time, prior_valid_time, prior_path_exp, depends_on=id)
# 1) Set posterior = prior
- id = prepare_IC_from_prior(prior_path_exp, prior_init_time, prior_valid_time, depends_on=id)
+ id = w.prepare_IC_from_prior(prior_path_exp, prior_init_time, prior_valid_time, depends_on=id)
# 2) Update posterior += updates from assimilation
- id = update_IC_from_DA(time, depends_on=id)
+ id = w.update_IC_from_DA(time, depends_on=id)
# How long shall we integrate?
timedelta_integrate = timedelta_btw_assim
@@ -70,7 +68,7 @@ if __name__ == "__main__":
output_restart_interval = 9999 # no restart file after last assim
# 3) Run WRF ensemble
- id = run_ENS(begin=time, # start integration from here
+ id = w.run_ENS(begin=time, # start integration from here
end=time + timedelta_integrate, # integrate until here
output_restart_interval=output_restart_interval,
depends_on=id)
@@ -78,7 +76,7 @@ if __name__ == "__main__":
# as we have WRF output, we can use own exp path as prior
prior_path_exp = cluster.archivedir
- id_sat = create_satimages(time, depends_on=id)
+ id_sat = w.create_satimages(time, depends_on=id)
# increment time
time += timedelta_btw_assim
@@ -86,6 +84,6 @@ if __name__ == "__main__":
# update time variables
prior_init_time = time - timedelta_btw_assim
- verify_sat(id_sat)
- verify_wrf(id)
- verify_fast(id)
+ w.verify_sat(id_sat)
+ w.verify_wrf(id)
+ w.verify_fast(id)
diff --git a/dartwrf/workflows.py b/dartwrf/workflows.py
index 9a09b02..89f614b 100644
--- a/dartwrf/workflows.py
+++ b/dartwrf/workflows.py
@@ -25,7 +25,7 @@ def dict_to_py(d, outfile):
class WorkFlows(object):
def __init__(self, exp_config='cfg.py', server_config='server.py'):
- """Set up the experiment folder in `archivedir`.
+ """Set up the experiment folder in `archivedir`, copy config files, backup scripts.
Args:
exp (str): Path to exp config file
@@ -240,7 +240,7 @@ class WorkFlows(object):
def create_satimages(self, init_time, depends_on=None):
cmd = self.cluster.python_verif+' ~/RTTOV-WRF/run_init.py '+self.cluster.archivedir+init_time.strftime('/%Y-%m-%d_%H:%M/')
- id = self.cluster.run_job(cmd, "RTTOV", cfg_update={"ntasks": "12", "time": "80", "mem": "200G"}, depends_on=[depends_on])
+ id = self.cluster.run_job(cmd, "RTTOV", cfg_update={"ntasks": "12", "time": "80", "mem": "180G"}, depends_on=[depends_on])
return id
--
GitLab