Commit 4ad99fa6 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

return empty if input is empty in dispatchers

parent 07635e1f
...@@ -263,6 +263,10 @@ class Dispatcher: ...@@ -263,6 +263,10 @@ class Dispatcher:
# unique obsid --> useless # unique obsid --> useless
# u_obsid = table.unique(data_basis["dataset", "obs_id"]) # u_obsid = table.unique(data_basis["dataset", "obs_id"])
# return an empty list if input is empty
if len(plan_basis) == 0 or len(data_basis) == 0:
return []
# initialize task list # initialize task list
task_list = [] task_list = []
...@@ -338,6 +342,10 @@ class Dispatcher: ...@@ -338,6 +342,10 @@ class Dispatcher:
) )
return sum(task_list, []) return sum(task_list, [])
# return an empty list if input is empty
if len(plan_basis) == 0 or len(data_basis) == 0:
return []
# unique obsid # unique obsid
u_obsid = table.unique(data_basis["dataset", "obs_id"]) u_obsid = table.unique(data_basis["dataset", "obs_id"])
relevant_plan = table.join( relevant_plan = table.join(
...@@ -450,6 +458,10 @@ class Dispatcher: ...@@ -450,6 +458,10 @@ class Dispatcher:
) )
return sum(task_list, []) return sum(task_list, [])
# return an empty list if input is empty
if len(plan_basis) == 0 or len(data_basis) == 0:
return []
# unique obsid # unique obsid
u_obsid = table.unique(data_basis["dataset", "obs_id"]) u_obsid = table.unique(data_basis["dataset", "obs_id"])
relevant_plan = table.join( relevant_plan = table.join(
...@@ -562,6 +574,10 @@ class Dispatcher: ...@@ -562,6 +574,10 @@ class Dispatcher:
data_basis: table.Table, data_basis: table.Table,
# n_jobs: int = 1, # n_jobs: int = 1,
): ):
# return an empty list if input is empty
if len(plan_basis) == 0 or len(data_basis) == 0:
return []
# unique obsgroup basis (using group_by) # unique obsgroup basis (using group_by)
obsgroup_basis = plan_basis.group_by( obsgroup_basis = plan_basis.group_by(
keys=[ keys=[
...@@ -652,6 +668,10 @@ class Dispatcher: ...@@ -652,6 +668,10 @@ class Dispatcher:
# n_jobs: int = 1, # n_jobs: int = 1,
) -> list[dict]: ) -> list[dict]:
# return an empty list if input is empty
if len(plan_basis) == 0 or len(data_basis) == 0:
return []
# unique obsgroup basis # unique obsgroup basis
obsgroup_basis = table.unique( obsgroup_basis = table.unique(
plan_basis[ plan_basis[
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment