Skip to content

Commit 8a9deb8

Browse files
committed
Fix the data input predictor of renamed samples.
1 parent 6e0e390 commit 8a9deb8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

graph_net/torch/constraint_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def __init__(self, config):
2121
self.config = config
2222

2323
def __call__(self, model_path, input_var_name: str) -> bool:
24-
return input_var_name.startswith("in_")
24+
return not input_var_name.startswith("w_")
2525

2626

2727
class ModelRunnablePredicator:

0 commit comments

Comments
 (0)