disable some check

Signed-off-by: yiliu30 <[email protected]>
intel · Jun 20, 2024 · 4b303f5 · 4b303f5
1 parent 256fc0a
commit 4b303f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neural_compressor/torch/algorithms/pt2e_quant/half_precision_rewriter.py b/neural_compressor/torch/algorithms/pt2e_quant/half_precision_rewriter.py
@@ -110,7 +110,7 @@ def is_target_node_in_candidate_list(match, original_graph, pattern_graph):
             if node.target == target_op:
                 target_node = node
                 break
-        if target_node is None:
+        if target_node is None:  # pragma: no cover
             return False
         matched_node = match.nodes_map[target_node]
         return matched_node in node_list
@@ -138,7 +138,7 @@ def get_unquantized_node_set(gm: torch.fx.GraphModule):
         if meta := getattr(node, "meta"):
             if quantization_annotation := meta.get(xiq.QUANT_ANNOTATION_KEY):
                 none_annotation = xiq._X86InductorQuantizationAnnotation(_annotated=True)
-                if quantization_annotation != none_annotation:
+                if quantization_annotation != none_annotation:  # pragma: no cover
                     continue
         unquantized_node_set.add(node)
     return unquantized_node_set