From b0440f6a10285284c333b28b0d82c7c0a04a4967 Mon Sep 17 00:00:00 2001
From: GZhao <zhaogangzz@gmail.com>
Date: Fri, 9 May 2025 14:26:40 +0800
Subject: [PATCH] update dataset api

---
 csst_cpic_sim/main.py            | 2 ++
 example/observations/05_sci.yaml | 1 +
 2 files changed, 3 insertions(+)

diff --git a/csst_cpic_sim/main.py b/csst_cpic_sim/main.py
index 7b6fd01..a2e27de 100644
--- a/csst_cpic_sim/main.py
+++ b/csst_cpic_sim/main.py
@@ -371,6 +371,7 @@ def observation_simulation_from_config(obs_file, config_file):
             time = obs_info.get('time', 0)
             emgain = obs_info.get('emgain', None)
             time = relative_time(time)
+            dataset = obs_info.get('dataset', 'default')
 
         except Exception as e:
             log.error(f"{file} is not a valid yaml file.")
@@ -416,6 +417,7 @@ def observation_simulation_from_config(obs_file, config_file):
                     output=output,
                     nsample=nsample,
                     csst_format=csst_format,
+                    dataset=dataset,
                     prograss_bar=True)
             except Exception as e:
                 log.error(f"{info_text} failed with {type(e).__name__}{e}.\n\n {traceback.format_exc()}")
diff --git a/example/observations/05_sci.yaml b/example/observations/05_sci.yaml
index b392407..acec337 100644
--- a/example/observations/05_sci.yaml
+++ b/example/observations/05_sci.yaml
@@ -5,6 +5,7 @@ expt: 100
 nframe: 2
 obsid: '40100000005'
 rotation: 30
+dataset: cpic-v01
 shift:
 - 0
 - 0
-- 
GitLab