From 150a0d8cc2ec4423aaa4d9c82181dc389267fbf4 Mon Sep 17 00:00:00 2001
From: wjm <wangjiming5@huawei.com>
Date: Fri, 11 Jun 2021 02:27:34 +0800
Subject: [PATCH] add

---
 parser/func_to_graph/func2graph.py                         | 2 +-
 .../parser/testcase/onnx_parser_testcase/onnx_model/if.py  | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/parser/func_to_graph/func2graph.py b/parser/func_to_graph/func2graph.py
index ef52916..633440f 100644
--- a/parser/func_to_graph/func2graph.py
+++ b/parser/func_to_graph/func2graph.py
@@ -1,4 +1,4 @@
-#!/usr/bin/python3
+#!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 #-------------------------------------------------------------------
 # Purpose:
diff --git a/tests/ut/parser/testcase/onnx_parser_testcase/onnx_model/if.py b/tests/ut/parser/testcase/onnx_parser_testcase/onnx_model/if.py
index e9aaef7..ae2caee 100644
--- a/tests/ut/parser/testcase/onnx_parser_testcase/onnx_model/if.py
+++ b/tests/ut/parser/testcase/onnx_parser_testcase/onnx_model/if.py
@@ -1,3 +1,10 @@
+#!/usr/bin/env python3
+# -*- coding: UTF-8 -*-
+#-------------------------------------------------------------------
+# Purpose:
+# Copyright 2021 Huawei Technologies Co., Ltd. All rights reserved.
+#-------------------------------------------------------------------
+
 # Given a bool scalar input cond.
 # return constant tensor x if cond is True, otherwise return constant tensor y.
 import numpy as np