"csst_dag/git@csst-tb.bao.ac.cn:csst-cicd/csst-dag.git" did not exist on "3ea74b496d163717bf4455c511450037852b8023"
Commit f77b60ac authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

assert data_basis and plan_basis non-zero

parent 6f2b7509
...@@ -197,9 +197,11 @@ class Dispatcher: ...@@ -197,9 +197,11 @@ class Dispatcher:
return data_basis return data_basis
@staticmethod @staticmethod
def find_plan_level0_basis(**kwargs) -> tuple(table.Table): def find_plan_level0_basis(**kwargs) -> tuple[table.Table]:
data_basis = Dispatcher.find_level0_basis(**kwargs) data_basis = Dispatcher.find_level0_basis(**kwargs)
plan_basis = Dispatcher.find_plan_basis(**kwargs) plan_basis = Dispatcher.find_plan_basis(**kwargs)
assert len(data_basis) > 0, data_basis
assert len(plan_basis) > 0, plan_basis
u_data_basis = table.unique(data_basis["dataset", "obs_id"]) u_data_basis = table.unique(data_basis["dataset", "obs_id"])
relevant_plan = table.join( relevant_plan = table.join(
u_data_basis, u_data_basis,
...@@ -207,12 +209,15 @@ class Dispatcher: ...@@ -207,12 +209,15 @@ class Dispatcher:
keys=["dataset", "obs_id"], keys=["dataset", "obs_id"],
join_type=PLAN_JOIN_TYPE, join_type=PLAN_JOIN_TYPE,
) )
assert len(relevant_plan) > 0, relevant_plan
return relevant_plan, data_basis return relevant_plan, data_basis
@staticmethod @staticmethod
def find_plan_level1_basis(**kwargs) -> tuple(table.Table): def find_plan_level1_basis(**kwargs) -> tuple[table.Table]:
data_basis = Dispatcher.find_level1_basis(**kwargs) data_basis = Dispatcher.find_level1_basis(**kwargs)
plan_basis = Dispatcher.find_plan_basis(**kwargs) plan_basis = Dispatcher.find_plan_basis(**kwargs)
assert len(data_basis) > 0, data_basis
assert len(plan_basis) > 0, plan_basis
u_data_basis = table.unique(data_basis["dataset", "obs_id"]) u_data_basis = table.unique(data_basis["dataset", "obs_id"])
relevant_plan = table.join( relevant_plan = table.join(
u_data_basis, u_data_basis,
...@@ -220,6 +225,7 @@ class Dispatcher: ...@@ -220,6 +225,7 @@ class Dispatcher:
keys=["dataset", "obs_id"], keys=["dataset", "obs_id"],
join_type=PLAN_JOIN_TYPE, join_type=PLAN_JOIN_TYPE,
) )
assert len(relevant_plan) > 0, relevant_plan
return relevant_plan, data_basis return relevant_plan, data_basis
@staticmethod @staticmethod
......
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