File tree 1 file changed +8
-4
lines changed
1 file changed +8
-4
lines changed Original file line number Diff line number Diff line change @@ -174,9 +174,9 @@ def __init__(
174
174
175
175
if log_system_metrics :
176
176
# Set system metrics sampling interval and samples before logging so that system metrics
177
- # are collected every 5s, and aggregated over 3 samples before being logged
178
- # (logging per 15s ).
179
- mlflow .set_system_metrics_samples_before_logging (3 )
177
+ # are collected every 5s, and aggregated over 6 samples before being logged
178
+ # (logging per 30s ).
179
+ mlflow .set_system_metrics_samples_before_logging (6 )
180
180
mlflow .set_system_metrics_sampling_interval (5 )
181
181
182
182
self ._rank_zero_only = rank_zero_only
@@ -545,7 +545,11 @@ def register_model_with_run_id(
545
545
"""
546
546
if self ._enabled :
547
547
from mlflow .exceptions import MlflowException
548
- from mlflow .protos .databricks_pb2 import ALREADY_EXISTS , RESOURCE_ALREADY_EXISTS , ErrorCode
548
+ from mlflow .protos .databricks_pb2 import (
549
+ ALREADY_EXISTS ,
550
+ RESOURCE_ALREADY_EXISTS ,
551
+ ErrorCode ,
552
+ )
549
553
550
554
full_name = f'{ self .model_registry_prefix } .{ name } ' if len (self .model_registry_prefix ) > 0 else name
551
555
You can’t perform that action at this time.
0 commit comments