Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
csst-cicd
csst-dag
Commits
f834e7cd
Commit
f834e7cd
authored
Jul 09, 2025
by
BO ZHANG
🏀
Browse files
fixed bug in dispatch_file
parent
028ad8b5
Changes
1
Hide whitespace changes
Inline
Side-by-side
tests/test_dispatcher.py
View file @
f834e7cd
...
...
@@ -4,29 +4,38 @@ from csst_dag.dag import Dispatcher
# # 1221 plan recs, 36630 data recs
plan_basis
,
data_basis
=
Dispatcher
.
load_test_data
()
print
(
len
(
plan_basis
),
len
(
data_basis
))
print
(
"plan conames: "
,
plan_basis
.
colnames
)
print
(
"data conames: "
,
data_basis
.
colnames
)
# # 666 task/s
task_list_via_file
=
Dispatcher
.
dispatch_file
(
plan_basis
,
data_basis
)
task_list_via_file
=
Dispatcher
.
dispatch_file
(
plan_basis
,
data_basis
[:
10
]
)
t
=
Table
(
task_list_via_file
)
np
.
unique
(
t
[
"success"
])
np
.
unique
(
t
[
"n_relevant_plan"
])
np
.
unique
(
t
[
"n_relevant_data"
])
print
(
t
[
"task"
])
print
(
sum
(
t
[
"success"
]))
print
(
task_list_via_file
[
0
][
"relevant_plan"
].
colnames
)
print
(
task_list_via_file
[
0
][
"relevant_data"
].
colnames
)
# # 13 task/s @n_jobs=1, 100*10 task/s @n_jobs=10 (max)
task_list_via_detector
=
Dispatcher
.
dispatch_detector
(
plan_basis
,
data_basis
[::
10
],
n_jobs
=
1
plan_basis
,
data_basis
[::
10
0
],
n_jobs
=
1
)
t
=
Table
(
task_list_via_detector
)
np
.
unique
(
t
[
"n_relevant_plan"
],
return_counts
=
True
)
np
.
unique
(
t
[
"success"
],
return_counts
=
True
)
print
(
task_list_via_detector
[
0
][
"relevant_plan"
].
colnames
)
print
(
task_list_via_detector
[
0
][
"relevant_data"
].
colnames
)
# 16 task/s @n_jobs=1, 130*10 tasks/s @n_jobs=10 (max) 🔼
task_list_via_obsid
=
Dispatcher
.
dispatch_obsid
(
plan_basis
,
data_basis
,
n_jobs
=
10
)
t
=
Table
(
task_list_via_obsid
)
np
.
unique
(
t
[
"n_relevant_plan"
],
return_counts
=
True
)
np
.
unique
(
t
[
"success"
],
return_counts
=
True
)
print
(
task_list_via_obsid
[
0
][
"relevant_plan"
].
colnames
)
print
(
task_list_via_obsid
[
0
][
"relevant_data"
].
colnames
)
# 13s/task
task_list_via_obsgroup
=
Dispatcher
.
dispatch_obsgroup
(
plan_basis
,
data_basis
)
...
...
@@ -36,3 +45,11 @@ print(
"/"
,
len
(
task_list_via_obsgroup
),
)
print
(
task_list_via_obsgroup
[
0
][
"relevant_plan"
].
colnames
)
print
(
task_list_via_obsgroup
[
0
][
"relevant_data"
].
colnames
)
# relevant plan_basis:
# ['dataset', 'instrument', 'obs_type', 'obs_group', 'obs_id', 'detector', 'n_frame', '_id']
# relevant data_basis:
# ['dataset', 'instrument', 'obs_type', 'obs_group', 'obs_id', 'detector', 'file_name', '_id']
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment