legacy mlclouds model as default for standard nsrdb runs

NREL · Jan 7, 2025 · 64d0ee8 · 64d0ee8
1 parent 6e255e0
commit 64d0ee8
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/nsrdb/config/create_configs.py b/nsrdb/config/create_configs.py
@@ -10,6 +10,8 @@
 import os
 import pprint
 
+from mlclouds import LEG_MODEL_FPATH
+
 from nsrdb.aggregation.aggregation import NSRDB_2018
 from nsrdb.config import (
     PIPELINE_CONFIG_TEMPLATE,
@@ -34,7 +36,19 @@
     'meta_dir': DEFAULT_META_DIR,
 }
 
-MAIN_KWARGS = {**BASE_KWARGS, 'extent': 'full', 'satellite': 'east'}
+DEFAULT_MLCLOUDS = {
+    'model_path': LEG_MODEL_FPATH,
+    'col_chunk': 10000,
+    'fill_all': False,
+    'max_workers': 4,
+}
+
+MAIN_KWARGS = {
+    **BASE_KWARGS,
+    'extent': 'full',
+    'satellite': 'east',
+    'ml-cloud-fill': DEFAULT_MLCLOUDS,
+}
 
 SURFRAD_KWARGS = {
     **MAIN_KWARGS,