62
62
from tensorboard .compat .proto import summary_pb2
63
63
from tensorboard .compat .proto import types_pb2
64
64
from tensorboard .plugins .graph import metadata as graph_metadata
65
- from tensorboard .uploader .proto import server_info_pb2
66
65
from tensorboard .util import tb_logging
67
66
from tensorboard .util import tensor_util
68
67
@@ -96,7 +95,7 @@ def __init__(
96
95
logdir : str ,
97
96
allowed_plugins : FrozenSet [str ],
98
97
experiment_display_name : Optional [str ] = None ,
99
- upload_limits : Optional [server_info_pb2 .UploadLimits ] = None ,
98
+ upload_limits : Optional [uploader_constants .UploadLimits ] = None ,
100
99
logdir_poll_rate_limiter : Optional [uploader_utils .RateLimiter ] = None ,
101
100
rpc_rate_limiter : Optional [uploader_utils .RateLimiter ] = None ,
102
101
tensor_rpc_rate_limiter : Optional [uploader_utils .RateLimiter ] = None ,
@@ -121,7 +120,7 @@ def __init__(
121
120
allowed_plugins: collection of string plugin names; events will only be
122
121
uploaded if their time series's metadata specifies one of these plugin
123
122
names
124
- upload_limits: instance of tensorboard.service. UploadLimits proto .
123
+ upload_limits: dataclass of uploader_constants. UploadLimits.
125
124
logdir_poll_rate_limiter: a `RateLimiter` to use to limit logdir polling
126
125
frequency, to avoid thrashing disks, especially on networked file
127
126
systems
@@ -160,29 +159,7 @@ def __init__(
160
159
161
160
self ._upload_limits = upload_limits
162
161
if not self ._upload_limits :
163
- self ._upload_limits = server_info_pb2 .UploadLimits ()
164
- self ._upload_limits .max_scalar_request_size = (
165
- uploader_constants .DEFAULT_MAX_SCALAR_REQUEST_SIZE
166
- )
167
- self ._upload_limits .min_scalar_request_interval = (
168
- uploader_constants .DEFAULT_MIN_SCALAR_REQUEST_INTERVAL
169
- )
170
- self ._upload_limits .min_tensor_request_interval = (
171
- uploader_constants .DEFAULT_MIN_TENSOR_REQUEST_INTERVAL
172
- )
173
- self ._upload_limits .max_tensor_request_size = (
174
- uploader_constants .DEFAULT_MAX_TENSOR_REQUEST_SIZE
175
- )
176
- self ._upload_limits .max_tensor_point_size = (
177
- uploader_constants .DEFAULT_MAX_TENSOR_POINT_SIZE
178
- )
179
- self ._upload_limits .min_blob_request_interval = (
180
- uploader_constants .DEFAULT_MIN_BLOB_REQUEST_INTERVAL
181
- )
182
- self ._upload_limits .max_blob_request_size = (
183
- uploader_constants .DEFAULT_MAX_BLOB_REQUEST_SIZE
184
- )
185
- self ._upload_limits .max_blob_size = uploader_constants .DEFAULT_MAX_BLOB_SIZE
162
+ self ._upload_limits = uploader_constants .UploadLimits ()
186
163
self ._description = description
187
164
self ._verbosity = verbosity
188
165
self ._one_shot = one_shot
@@ -482,7 +459,7 @@ def __init__(
482
459
experiment_resource_name : str ,
483
460
api : TensorboardServiceClient ,
484
461
allowed_plugins : Iterable [str ],
485
- upload_limits : server_info_pb2 .UploadLimits ,
462
+ upload_limits : uploader_constants .UploadLimits ,
486
463
rpc_rate_limiter : uploader_utils .RateLimiter ,
487
464
tensor_rpc_rate_limiter : uploader_utils .RateLimiter ,
488
465
blob_rpc_rate_limiter : uploader_utils .RateLimiter ,
0 commit comments