diff --git a/python/fastdeploy/__init__.py b/python/fastdeploy/__init__.py index bacad5bdf..b767393f1 100644 --- a/python/fastdeploy/__init__.py +++ b/python/fastdeploy/__init__.py @@ -37,4 +37,3 @@ from . import vision from . import pipeline from . import text from .download import download, download_and_decompress, download_model -from .utils import profile diff --git a/python/fastdeploy/utils/__init__.py b/python/fastdeploy/utils/__init__.py index 374afbad4..30fd1b3cb 100644 --- a/python/fastdeploy/utils/__init__.py +++ b/python/fastdeploy/utils/__init__.py @@ -11,6 +11,4 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. - -from .profile import profile from .example_resource import get_detection_test_image diff --git a/python/fastdeploy/utils/profile.py b/python/fastdeploy/utils/profile.py deleted file mode 100644 index 4c49cfe43..000000000 --- a/python/fastdeploy/utils/profile.py +++ /dev/null @@ -1,28 +0,0 @@ -# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - -import cProfile, pstats, io -from pstats import SortKey - - -def profile(func, *args, **kwargs): - pr = cProfile.Profile() - pr.enable() - func(*args, **kwargs) - pr.disable() - s = io.StringIO() - sortby = SortKey.CUMULATIVE - ps = pstats.Stats(pr, stream=s).sort_stats(sortby) - ps.print_stats() - print(s.getvalue())