def test_classify_no_affected():
        """
        If the outputs are from a classify op, it should not be affected by this graph pass.
        """

        @mb.program(input_specs=[mb.TensorSpec(shape=(3,))])
        def prog(x):
            classes = [np.int64(x) for x in range(3)]
            classes_var = mb.const(val=mil_list(classes))
            return mb.classify(probabilities=x, classes=classes_var)

        apply_pass_and_basic_check(prog, "mil_backend::adjust_io_to_supported_types")
        assert get_op_types_in_program(prog) == ["classify"]

        If the outputs are from a classify op, it should not be affected by this graph pass.
        
coremltools/converters/mil/backend/mil/passes/test_passes.py