Skip to content

Commit

Permalink
style: extend no-explicit-dtype check to xp and jnp
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz authored Oct 23, 2024
1 parent b4701da commit 83aaa37
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion source/checker/deepmd_checker.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ def visit_call(self, node):
if (
isinstance(node.func, Attribute)
and isinstance(node.func.expr, Name)
and node.func.expr.name in {"np", "tf", "torch"}
and node.func.expr.name in {"np", "tf", "torch", "xp", "jnp}
and node.func.attrname
in {
# https://pytorch.org/docs/stable/torch.html#creation-ops
Expand Down

0 comments on commit 83aaa37

Please sign in to comment.