Update unit tests

openvinotoolkit · Jun 27, 2023 · d893e1a · d893e1a
1 parent 53359e9
commit d893e1a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/algorithms/detection/adapters/mmdet/models/detectors/test_custom_dino_detector.py b/tests/unit/algorithms/detection/adapters/mmdet/models/detectors/test_custom_dino_detector.py
@@ -34,7 +34,7 @@ def test_custom_dino_load_state_pre_hook(self, fxt_cfg_custom_dino: Dict):
             "memory_trans_fc": "memory_trans_fc",
             "memory_trans_norm": "memory_trans_norm",
         }
-        model.load_state_dict_pre_hook(model, ckpt_dict)
+        model.load_state_dict_pre_hook(ckpt_dict)
 
         assert ckpt_dict["bbox_head.transformer.level_embeds"] == "level_embed"
         assert ckpt_dict["bbox_head.transformer.encoder.attentions.0"] == "encoder.self_attn"