提交 f30e7159 authored 作者: Michael Osthege's avatar Michael Osthege 提交者: Brandon T. Willard

Add type cast to resolve a mypy warning

上级 fd750a25
...@@ -18,6 +18,7 @@ from typing import ( ...@@ -18,6 +18,7 @@ from typing import (
Tuple, Tuple,
TypeVar, TypeVar,
Union, Union,
cast,
) )
import numpy as np import numpy as np
...@@ -1615,6 +1616,7 @@ def get_var_by_name( ...@@ -1615,6 +1616,7 @@ def get_var_by_name(
results: Tuple[Variable, ...] = () results: Tuple[Variable, ...] = ()
for var in walk(graphs, expand, False): for var in walk(graphs, expand, False):
var = cast(Variable, var)
if target_var_id == var.name or target_var_id == var.auto_name: if target_var_id == var.name or target_var_id == var.auto_name:
results += (var,) results += (var,)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论