Skip to content

Commit

Permalink
Data store: fix missing task proxy flow nums
Browse files Browse the repository at this point in the history
  • Loading branch information
MetRonnie committed Sep 11, 2024
1 parent a88b8d6 commit 94799a6
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions cylc/flow/data_store_mgr.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,6 +116,7 @@
if TYPE_CHECKING:
from cylc.flow.cycling import PointBase
from cylc.flow.flow_mgr import FlowNums
from cylc.flow.scheduler import Scheduler

EDGES = 'edges'
FAMILIES = 'families'
Expand Down Expand Up @@ -468,7 +469,7 @@ class DataStoreMgr:
ERR_PREFIX_JOB_NOT_ON_SEQUENCE = 'Invalid cycle point for job: '

def __init__(self, schd, n_edge_distance=1):
self.schd = schd
self.schd: Scheduler = schd
self.id_ = Tokens(
user=self.schd.owner,
workflow=self.schd.workflow,
Expand Down Expand Up @@ -1182,10 +1183,7 @@ def generate_ghost_task(
t_id = self.definition_id(name)

if itask is None:
itask = self.schd.pool.get_task(point_string, name)

if itask is None:
itask = TaskProxy(
itask = self.schd.pool.get_task(point_string, name) or TaskProxy(
self.id_,
self.schd.config.get_taskdef(name),
point,
Expand Down Expand Up @@ -1226,6 +1224,7 @@ def generate_ghost_task(
depth=task_def.depth,
graph_depth=n_depth,
name=name,
flow_nums=serialise_set(flow_nums),
)
self.all_n_window_nodes.add(tp_id)
self.n_window_depths.setdefault(n_depth, set()).add(tp_id)
Expand Down

0 comments on commit 94799a6

Please sign in to comment.