Skip to content

Commit

Permalink
Fix passing through the reduce_type in graph regression tasks
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 594226366
  • Loading branch information
Johannes Gasteiger authored and tensorflower-gardener committed Dec 28, 2023
1 parent e6b78bd commit 19c10f2
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions tensorflow_gnn/runner/tasks/regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -331,6 +331,7 @@ def __init__(self,
node_set_name,
units=units,
state_name=state_name,
reduce_type=reduce_type,
name=name,
label_fn=label_fn,
label_feature_name=label_feature_name)
Expand Down Expand Up @@ -369,6 +370,7 @@ def __init__(self,
node_set_name,
units=units,
state_name=state_name,
reduce_type=reduce_type,
name=name,
label_fn=label_fn,
label_feature_name=label_feature_name)
Expand Down Expand Up @@ -406,6 +408,7 @@ def __init__(self,
node_set_name,
units=units,
state_name=state_name,
reduce_type=reduce_type,
name=name,
label_fn=label_fn,
label_feature_name=label_feature_name)
Expand Down Expand Up @@ -444,6 +447,7 @@ def __init__(self,
node_set_name,
units=units,
state_name=state_name,
reduce_type=reduce_type,
name=name,
label_fn=label_fn,
label_feature_name=label_feature_name)
Expand Down Expand Up @@ -488,6 +492,7 @@ def __init__(self,
node_set_name,
units=units,
state_name=state_name,
reduce_type=reduce_type,
name=name,
label_fn=label_fn,
label_feature_name=label_feature_name,
Expand Down

0 comments on commit 19c10f2

Please sign in to comment.