add find_plan_level0_basis and find_plan_level1_basis (ced55eb3) · Commits · csst-cicd / csst-dag

csst_dag/dag/dispatcher.py→csst_dag/dag/_dispatcher.py

+26 −0

Original line number	Diff line number	Diff line
		@@ -196,6 +196,32 @@ class Dispatcher:
		)
		return data_basis

		@staticmethod
		def find_plan_level0_basis(**kwargs) -> tuple(table.Table):
		data_basis = Dispatcher.find_level0_basis(**kwargs)
		plan_basis = Dispatcher.find_plan_basis(**kwargs)
		u_data_basis = table.unique(data_basis["dataset", "obs_id"])
		relevant_plan = table.join(
		u_data_basis,
		plan_basis,
		keys=["dataset", "obs_id"],
		join_type=PLAN_JOIN_TYPE,
		)
		return relevant_plan, data_basis

		@staticmethod
		def find_plan_level1_basis(**kwargs) -> tuple(table.Table):
		data_basis = Dispatcher.find_level1_basis(**kwargs)
		plan_basis = Dispatcher.find_plan_basis(**kwargs)
		u_data_basis = table.unique(data_basis["dataset", "obs_id"])
		relevant_plan = table.join(
		u_data_basis,
		plan_basis,
		keys=["dataset", "obs_id"],
		join_type=PLAN_JOIN_TYPE,
		)
		return relevant_plan, data_basis

		@staticmethod
		def dispatch_file(
		plan_basis: table.Table,