Skip to content

Commit

Permalink
Telemetry fixes.
Browse files Browse the repository at this point in the history
  • Loading branch information
popovaan committed Nov 13, 2024
1 parent adeb3d2 commit 024d9d8
Show file tree
Hide file tree
Showing 5 changed files with 29 additions and 32 deletions.
14 changes: 7 additions & 7 deletions tools/ovc/openvino/tools/ovc/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
# SPDX-License-Identifier: Apache-2.0

from openvino.tools.ovc.convert import convert_model
from openvino.tools.ovc.telemetry_utils import is_optimum, init_mo_telemetry
from openvino.tools.ovc.telemetry_utils import is_optimum, init_ovc_telemetry

import importlib.metadata as importlib_metadata

Expand All @@ -11,10 +11,10 @@
except importlib_metadata.PackageNotFoundError:
optimum_version = None

from openvino.runtime import get_version as get_rt_version # pylint: disable=no-name-in-module,import-error
telemetry = init_ovc_telemetry('Model Conversion API')
telemetry.send_event("ov", "import", "general_import")

if is_optimum() and optimum_version is not None:
from openvino.runtime import get_version as get_rt_version # pylint: disable=no-name-in-module,import-error
telemetry = init_mo_telemetry("Optimum Intel", optimum_version)
telemetry.send_event("ov", "import", "import_from_optimum,ov_version:{}".format(get_rt_version()))
else:
telemetry = init_mo_telemetry()
telemetry.send_event("ov", "import", "general_import")
telemetry = init_ovc_telemetry("Optimum Intel", optimum_version)
telemetry.send_event("optimum", "import", "import_from_optimum,ov_version:{}".format(get_rt_version()))
4 changes: 2 additions & 2 deletions tools/ovc/openvino/tools/ovc/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import sys

from openvino.tools.ovc.main import main
from openvino.tools.ovc.telemetry_utils import init_mo_telemetry
from openvino.tools.ovc.telemetry_utils import init_ovc_telemetry

init_mo_telemetry()
init_ovc_telemetry()
sys.exit(main())
12 changes: 6 additions & 6 deletions tools/ovc/openvino/tools/ovc/convert_impl.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@
from openvino.tools.ovc.utils import check_values_equal
from openvino.tools.ovc.logger import init_logger
from openvino.tools.ovc.telemetry_utils import send_params_info, send_conversion_result, \
init_mo_telemetry
init_ovc_telemetry
from openvino.tools.ovc.moc_frontend.pytorch_frontend_utils import get_pytorch_decoder, \
extract_input_info_from_example, get_pytorch_decoder_for_model_on_disk
from openvino.tools.ovc.moc_frontend.paddle_frontend_utils import paddle_frontend_converter
Expand Down Expand Up @@ -428,12 +428,12 @@ def _convert(cli_parser: argparse.ArgumentParser, args, python_api_used):
tracemalloc.start()

simplified_ie_version = VersionChecker().get_ie_simplified_version()
telemetry = init_mo_telemetry()
telemetry = init_ovc_telemetry()
telemetry.start_session('ovc')
telemetry.send_event('ovc', 'version', simplified_ie_version)
# Initialize logger with 'ERROR' as default level to be able to form nice messages
# before arg parser deliver log_level requested by user
init_logger('ERROR', False)
#init_logger('ERROR', False)
argv = None
# Minimize modifications among other places in case if multiple pieces are passed as input_model
if python_api_used:
Expand Down Expand Up @@ -484,12 +484,12 @@ def _convert(cli_parser: argparse.ArgumentParser, args, python_api_used):

argv.feManager = FrontEndManager()

# send telemetry with params info
send_params_info(argv, cli_parser)

non_default_params = get_non_default_params(argv, cli_parser)
argv.is_python_api_used = python_api_used

# send telemetry with params info
send_params_info(non_default_params)

argv.framework = model_framework

orig_input_model = argv.input_model
Expand Down
4 changes: 2 additions & 2 deletions tools/ovc/openvino/tools/ovc/ovc.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,8 @@
import sys

if __name__ == "__main__":
from openvino.tools.ovc.telemetry_utils import init_mo_telemetry
from openvino.tools.ovc.telemetry_utils import init_ovc_telemetry
from openvino.tools.ovc.main import main

init_mo_telemetry()
init_ovc_telemetry()
sys.exit(main())
27 changes: 12 additions & 15 deletions tools/ovc/openvino/tools/ovc/telemetry_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ def is_optimum():
return False


def init_mo_telemetry(app_name='Model Conversion API', app_version=None):
def init_ovc_telemetry(app_name='OVC', app_version=None):
app_version = app_version if app_version is not None else get_rt_version()
return init_telemetry_class(tid=get_tid(),
app_name=app_name,
Expand Down Expand Up @@ -97,22 +97,19 @@ def arg_to_str(arg):
return str(type(arg))


def send_params_info(argv: argparse.Namespace, cli_parser: argparse.ArgumentParser):
def send_params_info(params: dict):
"""
This function sends information about used command line parameters.
:param argv: command line parameters.
:param cli_parser: command line parameters parser.
:param params: command line parameters dictionary.
"""
t = tm.Telemetry()
params_with_paths = get_params_with_paths_list()
for arg in vars(argv):
arg_value = getattr(argv, arg)
if not check_values_equal(arg_value, cli_parser.get_default(arg)):
if arg in params_with_paths:
# If command line argument value is a directory or a path to file it is not sent
# as it may contain confidential information. "1" value is used instead.
param_str = arg + ":" + str(1)
else:
param_str = arg + ":" + arg_to_str(arg_value)

t.send_event('ovc', 'cli_parameters', param_str)
for key, value in params.items():
if key in params_with_paths:
# If command line argument value is a directory or a path to file it is not sent
# as it may contain confidential information. "1" value is used instead.
param_str = key + ":" + str(1)
else:
param_str = key + ":" + arg_to_str(value)

t.send_event('ovc', 'cli_parameters', param_str)

0 comments on commit 024d9d8

Please sign in to comment.