Fix model reference for pylint (#4338)

19e737c9 · Yuge Zhang · GitHub · 5ce474bc · 19e737c9
Unverified Commit 19e737c9 authored Nov 26, 2021 by Yuge Zhang Committed by GitHub Nov 26, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

nni/retiarii/graph.py nni/retiarii/graph.py +1 -1

No files found.
--- a/nni/retiarii/graph.py
+++ b/nni/retiarii/graph.py
@@ -114,7 +114,7 @@ class Model:
        self.graphs: Dict[str, Graph] = {}
        self.evaluator: Optional[Evaluator] = None

-        self.history: List[Model] = []
+        self.history: List['Model'] = []

        self.metric: Optional[MetricData] = None
        self.intermediate_metrics: List[MetricData] = []