Skip to content

Tags not propogating to Endpoints and Models #586

@AnshumanRanjan

Description

@AnshumanRanjan

Please fill out the form below.

System Information

  • Framework (e.g. TensorFlow) / Algorithm (e.g. KMeans): Sklearn
  • Framework Version:NA
  • Python Version: 3.6.6
  • CPU or GPU: CPU
  • Python SDK Version:1.16.3
  • Are you using a custom image: NO

Describe the problem

Based on my testing , tags parameter defined in the Estimator objects are not propagated to Models and Endpoints . Although they do get propagated to Training jobs .

The other thing we notice is that things like subnets and security group do get propogated from Estimator object to Endpoints and models

  • Exact command to reproduce:
    Using sample notebook and modifying :
    notebooks/scikit_learn_inference_pipeline_2019-01-07/Inference Pipeline with Scikit-learn and Linear Learner.ipynb
from sagemaker.sklearn.estimator import SKLearn
script_path = 'sklearn_abalone_featurizer.py'
tagtobeadded = [{'Key':'TagtestKey','Value':'TagtestValue'}]
sklearn_preprocessor = SKLearn(
    entry_point=script_path,
    role=role,
    tags=tagtobeadded,
    train_instance_type="ml.c4.xlarge",
    sagemaker_session=sagemaker_session)

The below training Job will have tags propogated
sklearn_preprocessor.fit({'train': train_input})

The below Endpoint and Model created will not have the tag information

endpoint_name = 'TagsTest'
sklearn_preprocessor.deploy(initial_instance_count=1, instance_type='ml.c4.large', endpoint_name=endpoint_name)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions