From a24f94a36c40c3ce5d5fb07771b8ace9cec47dd8 Mon Sep 17 00:00:00 2001 From: KUANGDD Date: Wed, 16 Oct 2024 01:14:51 +0800 Subject: [PATCH] remove useless codes Former-commit-id: 9b2642a2b53d3392e95061ed0f2c8dc10580c9e8 --- tests/data/test_mm_plugin.py | 36 ------------------------------------ 1 file changed, 36 deletions(-) diff --git a/tests/data/test_mm_plugin.py b/tests/data/test_mm_plugin.py index 32b89f7f..b342e658 100644 --- a/tests/data/test_mm_plugin.py +++ b/tests/data/test_mm_plugin.py @@ -68,42 +68,6 @@ def _get_mm_inputs(processor: "ProcessorMixin") -> Dict[str, "torch.Tensor"]: image_processor: "BaseImageProcessor" = getattr(processor, "image_processor") return image_processor(images=IMAGES, return_tensors="pt") -def _is_nested_tensor_list(element): - if not isinstance(element, list): - return False - - for item in element: - if isinstance(item, list): - if not _is_nested_tensor_list(item): - return False - - elif not isinstance(item, torch.Tensor): - return False - - return True - - -def _equal_nested_tensor_list(a: List[List[torch.Tensor]], b: List[List[torch.Tensor]]) -> bool: - if type(a) != type(b): - return False - - if isinstance(a, list) and isinstance(b, list): - if len(a) != len(b): - return False - - for sub_a, sub_b in zip(a, b): - if isinstance(sub_a, list) and isinstance(sub_b, list): - if not _equal_nested_tensor_list(sub_a, sub_b): - return False - elif isinstance(sub_a, torch.Tensor) and isinstance(sub_b, torch.Tensor): - if not torch.equal(sub_a, sub_b): - return False - else: - return False - - return True - - return False def _is_close(batch_a: Dict[str, Any], batch_b: Dict[str, Any]) -> None: assert batch_a.keys() == batch_b.keys()