@@ -64,33 +64,33 @@ sagemaker_featurestore_runtime_client = boto3.Session().client(
6464### OPTION 1: To Download all records at once and upload records sequentially
6565with open('./Downloads/Sample_data.csv') as file_handle:
6666 records =[
67- [
68- {'FeatureName':featureName,
69- 'ValueAsString':valueAsString}
70- for featureName, valueAsString in row.items()]
71- for row in csv.DictReader(file_handle, skipinitialspace=True)]
72-
67+ [
68+ {'FeatureName':featureName,
69+ 'ValueAsString':valueAsString}
70+ for featureName, valueAsString in row.items()]
71+ for row in csv.DictReader(file_handle, skipinitialspace=True)]
72+
7373for record in records:
7474 sagemaker_featurestore_runtime_client.put_record(
75- FeatureGroupName=feature_group_name,
76- Record=record)
75+ FeatureGroupName=feature_group_name,
76+ Record=record)
7777
7878### OPTION 2: To Download records sequentially and upload records sequentially
7979with open('./Downloads/Sample_data.csv') as file_handle:
8080 for row in csv.DictReader(file_handle, skipinitialspace=True):
81- record =[
82- {'FeatureName':featureName,
83- 'ValueAsString':valueAsString}
84- for featureName, valueAsString in row.items()]
85- sagemaker_featurestore_runtime_client.put_record(
86- FeatureGroupName=feature_group_name,
87- Record=record)
88-
89- # To Check that the records are retrievable
81+ record =[
82+ {'FeatureName':featureName,
83+ 'ValueAsString':valueAsString}
84+ for featureName, valueAsString in row.items()]
85+ sagemaker_featurestore_runtime_client.put_record(
86+ FeatureGroupName=feature_group_name,
87+ Record=record)
88+
89+ # To Check that the records are retrievable
9090for recordIdentifierValue in range(1,len(records) + 1):
9191 sagemaker_featurestore_runtime_client.get_record(
92- FeatureGroupName=feature_group_name,
93- RecordIdentifierValueAsString=str(recordIdentifierValue))
92+ FeatureGroupName=feature_group_name,
93+ RecordIdentifierValueAsString=str(recordIdentifierValue))
9494```
9595
9696## Deleting your Feature Group
0 commit comments