Skip to content

Instantly share code, notes, and snippets.

@chrisking
Created February 19, 2020 21:25
Show Gist options
  • Select an option

  • Save chrisking/8d92b9e040e4da465958d65eec2584fc to your computer and use it in GitHub Desktop.

Select an option

Save chrisking/8d92b9e040e4da465958d65eec2584fc to your computer and use it in GitHub Desktop.
meta_schema_val = [{"AttributeName": "item_id", "AttributeType": "string"},
{"AttributeName": "style", "AttributeType": "string"},
{"AttributeName": "category", "AttributeType": "string"},
{"AttributeName": "subcat", "AttributeType": "string"}]
meta_schema = {"Attributes": meta_schema_val}
response = forecast.create_dataset(Domain="CUSTOM",
DatasetType='ITEM_METADATA',
DatasetName=meta_dataset_name,
Schema=meta_schema
)
meta_dataset_arn = response['DatasetArn']
dataset_arns = []
dataset_arns.append(ts_dataset_arn)
dataset_arns.append(meta_dataset_arn)
forecast.update_dataset_group(DatasetGroupArn=dataset_group_arn, DatasetArns=dataset_arns)
meta_s3_data_path = f"{s3_data_path}/metadata.csv"
meta_dataset_import_job_response = forecast.create_dataset_import_job(DatasetImportJobName=dataset_group,
DatasetArn=meta_dataset_arn,
DataSource= {
"S3Config" : {
"Path": meta_s3_data_path,
"RoleArn": role_arn
}
})
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment