natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed Telemetry initialization. (#22464)

This commit is contained in:
Anastasiia Pnevskaia 2024-01-30 10:18:36 +01:00 committed by GitHub
parent 3afad96899
commit e2533e8126
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 69 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
tools/mo/openvino/tools/mo/utils/telemetry_utils.py
View File

@ -22,13 +22,41 @@ except ImportError:
def init_mo_telemetry(app_name='Model Optimizer'):
return tm.Telemetry(tid=get_tid(),
app_name=app_name,
app_version=get_rt_version(),
backend='ga4',
enable_opt_in_dialog=False,
disable_in_ci=True
)
return init_telemetry_class(tid=get_tid(),
app_name=app_name,
app_version=get_rt_version(),
backend='ga4',
enable_opt_in_dialog=False,
disable_in_ci=True
)
def init_telemetry_class(tid,
app_name,
app_version,
backend,
enable_opt_in_dialog,
disable_in_ci):
# Init telemetry class
telemetry = tm.Telemetry(tid=tid,
app_name=app_name,
app_version=app_version,
backend=backend,
enable_opt_in_dialog=enable_opt_in_dialog,
disable_in_ci=disable_in_ci)
# Telemetry is a singleton class and if it was already initialized in another tool
# some parameters will be incorrect, including app_name.
# In this case we need to force reinitialisation of telemetry.
if hasattr(telemetry, "backend") and telemetry.backend.app_name != app_name:
telemetry.init(tid=tid,
app_name=app_name,
app_version=app_version,
backend=backend,
enable_opt_in_dialog=enable_opt_in_dialog,
disable_in_ci=disable_in_ci)
return telemetry
def send_framework_info(framework: str):
"""

41
tools/ovc/openvino/tools/ovc/telemetry_utils.py
View File

@ -17,13 +17,40 @@ except ImportError:
def init_mo_telemetry(app_name='Model Conversion API'):
return tm.Telemetry(tid=get_tid(),
app_name=app_name,
app_version=get_rt_version(),
backend='ga4',
enable_opt_in_dialog=False,
disable_in_ci=True
)
return init_telemetry_class(tid=get_tid(),
app_name=app_name,
app_version=get_rt_version(),
backend='ga4',
enable_opt_in_dialog=False,
disable_in_ci=True
)
def init_telemetry_class(tid,
app_name,
app_version,
backend,
enable_opt_in_dialog,
disable_in_ci):
# Init telemetry class
telemetry = tm.Telemetry(tid=tid,
app_name=app_name,
app_version=app_version,
backend=backend,
enable_opt_in_dialog=enable_opt_in_dialog,
disable_in_ci=disable_in_ci)
# Telemetry is a singleton class and if it was already initialized in another tool
# some parameters will be incorrect, including app_name.
# In this case we need to force reinitialisation of telemetry.
if hasattr(telemetry, "backend") and telemetry.backend.app_name != app_name:
telemetry.init(tid=tid,
app_name=app_name,
app_version=app_version,
backend=backend,
enable_opt_in_dialog=enable_opt_in_dialog,
disable_in_ci=disable_in_ci)
return telemetry
def send_framework_info(framework: str):