From db0ab00a1b29c3d7726ffd1d71ff44c793aa1a38 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Thu, 6 Apr 2023 19:31:02 +0800
Subject: [PATCH] fix(imperative/functional): fix general-norm assert always be
 true

GitOrigin-RevId: f8d4da570f66f36f3c743e67a1a1c1222f33aad8
---
 imperative/python/megengine/functional/nn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py
index 4bed60525..2e1a6221f 100644
--- a/imperative/python/megengine/functional/nn.py
+++ b/imperative/python/megengine/functional/nn.py
@@ -1188,7 +1188,7 @@ def general_norm(
         sorted(normalized_axis), normalized_axis
     )
     assert (
-        normalized_axis[-1] < inp.ndim,
+        normalized_axis[-1] < inp.ndim
     ), "the maximum axis in normalized_axis is greater than inp_shape.ndim"
     assert len(set(normalized_axis)) == len(
         normalized_axis
-- 
GitLab