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