From 6a453167e82cd0dc4877271aa3ca86324a61b54e Mon Sep 17 00:00:00 2001 From: liukai Date: Thu, 29 Sep 2022 20:12:20 +0800 Subject: [PATCH] fix bug --- .../test_mutables/test_units/test_mutable_channel_units.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models/test_mutables/test_units/test_mutable_channel_units.py b/tests/test_models/test_mutables/test_units/test_mutable_channel_units.py index 7154bc518..a0c97b6a1 100644 --- a/tests/test_models/test_mutables/test_units/test_mutable_channel_units.py +++ b/tests/test_models/test_mutables/test_units/test_mutable_channel_units.py @@ -121,7 +121,7 @@ def test_replace_with_dynamic_ops(self): for unit_type in GROUPS: with self.subTest(model=model_data, unit=unit_type): model: nn.Module = model_data() - graph = ModuleGraph.init_from_fx_tracer(model) + graph = ModuleGraph.init_from_backward_tracer(model) units: List[ MutableChannelUnit] = unit_type.init_from_graph( graph)