remove np

susanbao · susanbao · commit f8dd1770d04d · 2025-10-09T08:53:32.000Z
diff --git a/src/maxdiffusion/tests/wan_transformer_test.py b/src/maxdiffusion/tests/wan_transformer_test.py
@@ -38,7 +38,6 @@
 from maxdiffusion.pyconfig import HyperParameters
 from maxdiffusion.pipelines.wan.wan_pipeline import WanPipeline
 import qwix
-import numpy as np
 from flax.linen import partitioning as nn_partitioning
 
 RealQtRule = qwix.QtRule
