Skip to content

Commit a61d042

Browse files
committed
added telemetry tracking
1 parent 60a421d commit a61d042

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/sagemaker/remote_function/client.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,12 +35,13 @@
3535

3636
from sagemaker.session import Session
3737
from sagemaker.s3 import s3_path_join
38-
from sagemaker.remote_function.job import _Job, _JobSettings
39-
from sagemaker.remote_function.job import _RunInfo
38+
from sagemaker.remote_function.job import _JobSettings, _Job, _RunInfo
4039
from sagemaker.remote_function import logging_config
4140
from sagemaker.utils import name_from_base, base_from_name
4241
from sagemaker.remote_function.spark_config import SparkConfig
4342
from sagemaker.remote_function.custom_file_filter import CustomFileFilter
43+
from sagemaker.telemetry.telemetry_logging import _telemetry_emitter
44+
from sagemaker.telemetry.constants import Feature
4445

4546
_API_CALL_LIMIT = {
4647
"SubmittingIntervalInSecs": 1,
@@ -57,7 +58,7 @@
5758

5859
logger = logging_config.get_logger()
5960

60-
61+
@_telemetry_emitter(feature=Feature.REMOTE_FUNCTION, func_name="remote_function.remote")
6162
def remote(
6263
_func=None,
6364
*,

0 commit comments

Comments
 (0)