-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Closed
Description
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
Labels
No labels