Skip to content

Commit 6e5d91d

Browse files
mollyheamazonmollyheamazon
authored
Merge branch 'aws:master-v2' into master-v2
2 parents fc98562 + be0d6b5 commit 6e5d91d

12 files changed

Lines changed: 406 additions & 7 deletions

File tree

‎CHANGELOG.md‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,10 @@
11
# Changelog
22

3+
## v2.257.0 (2026-02-03)
4+
5+
### Features
6+
* Update image URIs for DJL 0.36.0 release
7+
38
## v2.256.1 (2026-01-21)
49

510
### Bug fixes and Other Changes

‎VERSION‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
2.256.1
1+
2.257.0

‎src/sagemaker/feature_store/feature_group.py‎

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,8 @@
7171
TargetStoreEnum,
7272
)
7373
from sagemaker.utils import resolve_value_from_config, format_tags, Tags
74+
from sagemaker.telemetry.telemetry_logging import _telemetry_emitter
75+
from sagemaker.telemetry.constants import Feature
7476

7577
logger = logging.getLogger(__name__)
7678

@@ -672,6 +674,7 @@ class FeatureGroup:
672674
FeatureTypeEnum.STRING.value: "STRING",
673675
}
674676

677+
@_telemetry_emitter(feature=Feature.FEATURE_STORE_V2, func_name="feature_group.create")
675678
def create(
676679
self,
677680
s3_uri: Union[str, bool],
@@ -788,10 +791,12 @@ def create(
788791

789792
return self.sagemaker_session.create_feature_group(**create_feature_store_args)
790793

794+
@_telemetry_emitter(feature=Feature.FEATURE_STORE_V2, func_name="feature_group.delete")
791795
def delete(self):
792796
"""Delete a FeatureGroup."""
793797
self.sagemaker_session.delete_feature_group(feature_group_name=self.name)
794798

799+
@_telemetry_emitter(feature=Feature.FEATURE_STORE_V2, func_name="feature_group.describe")
795800
def describe(self, next_token: str = None) -> Dict[str, Any]:
796801
"""Describe a FeatureGroup.
797802
@@ -805,6 +810,7 @@ def describe(self, next_token: str = None) -> Dict[str, Any]:
805810
feature_group_name=self.name, next_token=next_token
806811
)
807812

813+
@_telemetry_emitter(feature=Feature.FEATURE_STORE_V2, func_name="feature_group.update")
808814
def update(
809815
self,
810816
feature_additions: Sequence[FeatureDefinition] = None,
@@ -843,6 +849,9 @@ def update(
843849
throughput_config=throughput_config_parameter,
844850
)
845851

852+
@_telemetry_emitter(
853+
feature=Feature.FEATURE_STORE_V2, func_name="feature_group.update_feature_metadata"
854+
)
846855
def update_feature_metadata(
847856
self,
848857
feature_name: str,
@@ -871,6 +880,9 @@ def update_feature_metadata(
871880
parameter_removals=(parameter_removals or []),
872881
)
873882

883+
@_telemetry_emitter(
884+
feature=Feature.FEATURE_STORE_V2, func_name="feature_group.describe_feature_metadata"
885+
)
874886
def describe_feature_metadata(self, feature_name: str) -> Dict[str, Any]:
875887
"""Describe feature metadata by feature name.
876888
@@ -1038,6 +1050,7 @@ def load_feature_definitions(
10381050
self.feature_definitions = feature_definitions
10391051
return self.feature_definitions
10401052

1053+
@_telemetry_emitter(feature=Feature.FEATURE_STORE_V2, func_name="feature_group.get_record")
10411054
def get_record(
10421055
self,
10431056
record_identifier_value_as_string: str,
@@ -1057,6 +1070,7 @@ def get_record(
10571070
feature_names=feature_names,
10581071
).get("Record")
10591072

1073+
@_telemetry_emitter(feature=Feature.FEATURE_STORE_V2, func_name="feature_group.put_record")
10601074
def put_record(
10611075
self,
10621076
record: Sequence[FeatureValue],
@@ -1080,6 +1094,7 @@ def put_record(
10801094
ttl_duration=ttl_duration.to_dict() if ttl_duration is not None else None,
10811095
)
10821096

1097+
@_telemetry_emitter(feature=Feature.FEATURE_STORE_V2, func_name="feature_group.delete_record")
10831098
def delete_record(
10841099
self,
10851100
record_identifier_value_as_string: str,

‎src/sagemaker/feature_store/feature_store.py‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,8 @@
3333
SortOrderEnum,
3434
Identifier,
3535
)
36+
from sagemaker.telemetry.telemetry_logging import _telemetry_emitter
37+
from sagemaker.telemetry.constants import Feature
3638

3739

3840
@attr.s
@@ -47,6 +49,7 @@ class FeatureStore:
4749

4850
sagemaker_session: Session = attr.ib(default=Session)
4951

52+
@_telemetry_emitter(feature=Feature.FEATURE_STORE_V2, func_name="feature_store.create_dataset")
5053
def create_dataset(
5154
self,
5255
base: Union[FeatureGroup, pd.DataFrame],
@@ -92,6 +95,9 @@ def create_dataset(
9295
kms_key_id,
9396
)
9497

98+
@_telemetry_emitter(
99+
feature=Feature.FEATURE_STORE_V2, func_name="feature_store.list_feature_groups"
100+
)
95101
def list_feature_groups(
96102
self,
97103
name_contains: str = None,
@@ -137,6 +143,9 @@ def list_feature_groups(
137143
next_token=next_token,
138144
)
139145

146+
@_telemetry_emitter(
147+
feature=Feature.FEATURE_STORE_V2, func_name="feature_store.batch_get_record"
148+
)
140149
def batch_get_record(
141150
self,
142151
identifiers: Sequence[Identifier],

‎src/sagemaker/image_uri_config/djl-lmi.json‎

Lines changed: 253 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,261 @@
33
"inference"
44
],
55
"version_aliases": {
6-
"latest": "0.30.0"
6+
"latest": "0.36.0"
77
},
88
"versions": {
9+
"0.36.0": {
10+
"registries": {
11+
"af-south-1": "626614931356",
12+
"il-central-1": "780543022126",
13+
"ap-east-1": "871362719292",
14+
"ap-northeast-1": "763104351884",
15+
"ap-northeast-2": "763104351884",
16+
"ap-northeast-3": "364406365360",
17+
"ap-south-1": "763104351884",
18+
"ap-southeast-1": "763104351884",
19+
"ap-southeast-2": "763104351884",
20+
"ap-southeast-3": "907027046896",
21+
"ca-central-1": "763104351884",
22+
"cn-north-1": "727897471807",
23+
"cn-northwest-1": "727897471807",
24+
"eu-central-1": "763104351884",
25+
"eu-north-1": "763104351884",
26+
"eu-west-1": "763104351884",
27+
"eu-west-2": "763104351884",
28+
"eu-west-3": "763104351884",
29+
"eu-south-1": "692866216735",
30+
"me-south-1": "217643126080",
31+
"me-central-1": "914824155844",
32+
"sa-east-1": "763104351884",
33+
"us-east-1": "763104351884",
34+
"us-east-2": "763104351884",
35+
"us-gov-east-1": "446045086412",
36+
"us-gov-west-1": "442386744353",
37+
"us-west-1": "763104351884",
38+
"us-west-2": "763104351884",
39+
"ca-west-1": "204538143572",
40+
"eu-central-2": "380420809688",
41+
"ap-south-2": "772153158452",
42+
"eu-south-2": "503227376785",
43+
"ap-southeast-7": "590183813437",
44+
"ap-southeast-4": "457447274322",
45+
"ap-southeast-5": "550225433462",
46+
"mx-central-1": "637423239942"
47+
},
48+
"repository": "djl-inference",
49+
"tag_prefix": "0.36.0-lmi18.0.0-cu128"
50+
},
51+
"0.35.0": {
52+
"registries": {
53+
"af-south-1": "626614931356",
54+
"il-central-1": "780543022126",
55+
"ap-east-1": "871362719292",
56+
"ap-northeast-1": "763104351884",
57+
"ap-northeast-2": "763104351884",
58+
"ap-northeast-3": "364406365360",
59+
"ap-south-1": "763104351884",
60+
"ap-southeast-1": "763104351884",
61+
"ap-southeast-2": "763104351884",
62+
"ap-southeast-3": "907027046896",
63+
"ca-central-1": "763104351884",
64+
"cn-north-1": "727897471807",
65+
"cn-northwest-1": "727897471807",
66+
"eu-central-1": "763104351884",
67+
"eu-north-1": "763104351884",
68+
"eu-west-1": "763104351884",
69+
"eu-west-2": "763104351884",
70+
"eu-west-3": "763104351884",
71+
"eu-south-1": "692866216735",
72+
"me-south-1": "217643126080",
73+
"me-central-1": "914824155844",
74+
"sa-east-1": "763104351884",
75+
"us-east-1": "763104351884",
76+
"us-east-2": "763104351884",
77+
"us-gov-east-1": "446045086412",
78+
"us-gov-west-1": "442386744353",
79+
"us-west-1": "763104351884",
80+
"us-west-2": "763104351884",
81+
"ca-west-1": "204538143572",
82+
"eu-central-2": "380420809688",
83+
"ap-south-2": "772153158452",
84+
"eu-south-2": "503227376785",
85+
"ap-southeast-7": "590183813437",
86+
"ap-southeast-4": "457447274322",
87+
"ap-southeast-5": "550225433462",
88+
"mx-central-1": "637423239942"
89+
},
90+
"repository": "djl-inference",
91+
"tag_prefix": "0.35.0-lmi17.0.0-cu128"
92+
},
93+
"0.34.0": {
94+
"registries": {
95+
"af-south-1": "626614931356",
96+
"il-central-1": "780543022126",
97+
"ap-east-1": "871362719292",
98+
"ap-northeast-1": "763104351884",
99+
"ap-northeast-2": "763104351884",
100+
"ap-northeast-3": "364406365360",
101+
"ap-south-1": "763104351884",
102+
"ap-southeast-1": "763104351884",
103+
"ap-southeast-2": "763104351884",
104+
"ap-southeast-3": "907027046896",
105+
"ca-central-1": "763104351884",
106+
"cn-north-1": "727897471807",
107+
"cn-northwest-1": "727897471807",
108+
"eu-central-1": "763104351884",
109+
"eu-north-1": "763104351884",
110+
"eu-west-1": "763104351884",
111+
"eu-west-2": "763104351884",
112+
"eu-west-3": "763104351884",
113+
"eu-south-1": "692866216735",
114+
"me-south-1": "217643126080",
115+
"me-central-1": "914824155844",
116+
"sa-east-1": "763104351884",
117+
"us-east-1": "763104351884",
118+
"us-east-2": "763104351884",
119+
"us-gov-east-1": "446045086412",
120+
"us-gov-west-1": "442386744353",
121+
"us-west-1": "763104351884",
122+
"us-west-2": "763104351884",
123+
"ca-west-1": "204538143572",
124+
"eu-central-2": "380420809688",
125+
"ap-south-2": "772153158452",
126+
"eu-south-2": "503227376785",
127+
"ap-southeast-7": "590183813437",
128+
"ap-southeast-4": "457447274322",
129+
"ap-southeast-5": "550225433462",
130+
"mx-central-1": "637423239942"
131+
},
132+
"repository": "djl-inference",
133+
"tag_prefix": "0.34.0-lmi16.0.0-cu128"
134+
},
135+
"0.33.0": {
136+
"registries": {
137+
"af-south-1": "626614931356",
138+
"il-central-1": "780543022126",
139+
"ap-east-1": "871362719292",
140+
"ap-northeast-1": "763104351884",
141+
"ap-northeast-2": "763104351884",
142+
"ap-northeast-3": "364406365360",
143+
"ap-south-1": "763104351884",
144+
"ap-southeast-1": "763104351884",
145+
"ap-southeast-2": "763104351884",
146+
"ap-southeast-3": "907027046896",
147+
"ca-central-1": "763104351884",
148+
"cn-north-1": "727897471807",
149+
"cn-northwest-1": "727897471807",
150+
"eu-central-1": "763104351884",
151+
"eu-north-1": "763104351884",
152+
"eu-west-1": "763104351884",
153+
"eu-west-2": "763104351884",
154+
"eu-west-3": "763104351884",
155+
"eu-south-1": "692866216735",
156+
"me-south-1": "217643126080",
157+
"me-central-1": "914824155844",
158+
"sa-east-1": "763104351884",
159+
"us-east-1": "763104351884",
160+
"us-east-2": "763104351884",
161+
"us-gov-east-1": "446045086412",
162+
"us-gov-west-1": "442386744353",
163+
"us-west-1": "763104351884",
164+
"us-west-2": "763104351884",
165+
"ca-west-1": "204538143572",
166+
"eu-central-2": "380420809688",
167+
"ap-south-2": "772153158452",
168+
"eu-south-2": "503227376785",
169+
"ap-southeast-7": "590183813437",
170+
"ap-southeast-4": "457447274322",
171+
"ap-southeast-5": "550225433462",
172+
"mx-central-1": "637423239942"
173+
},
174+
"repository": "djl-inference",
175+
"tag_prefix": "0.33.0-lmi15.0.0-cu128"
176+
},
177+
"0.32.0": {
178+
"registries": {
179+
"af-south-1": "626614931356",
180+
"il-central-1": "780543022126",
181+
"ap-east-1": "871362719292",
182+
"ap-northeast-1": "763104351884",
183+
"ap-northeast-2": "763104351884",
184+
"ap-northeast-3": "364406365360",
185+
"ap-south-1": "763104351884",
186+
"ap-southeast-1": "763104351884",
187+
"ap-southeast-2": "763104351884",
188+
"ap-southeast-3": "907027046896",
189+
"ca-central-1": "763104351884",
190+
"cn-north-1": "727897471807",
191+
"cn-northwest-1": "727897471807",
192+
"eu-central-1": "763104351884",
193+
"eu-north-1": "763104351884",
194+
"eu-west-1": "763104351884",
195+
"eu-west-2": "763104351884",
196+
"eu-west-3": "763104351884",
197+
"eu-south-1": "692866216735",
198+
"me-south-1": "217643126080",
199+
"me-central-1": "914824155844",
200+
"sa-east-1": "763104351884",
201+
"us-east-1": "763104351884",
202+
"us-east-2": "763104351884",
203+
"us-gov-east-1": "446045086412",
204+
"us-gov-west-1": "442386744353",
205+
"us-west-1": "763104351884",
206+
"us-west-2": "763104351884",
207+
"ca-west-1": "204538143572",
208+
"eu-central-2": "380420809688",
209+
"ap-south-2": "772153158452",
210+
"eu-south-2": "503227376785",
211+
"ap-southeast-7": "590183813437",
212+
"ap-southeast-4": "457447274322",
213+
"ap-southeast-5": "550225433462",
214+
"mx-central-1": "637423239942"
215+
},
216+
"repository": "djl-inference",
217+
"tag_prefix": "0.32.0-lmi14.0.0-cu126"
218+
},
219+
"0.31.0": {
220+
"registries": {
221+
"af-south-1": "626614931356",
222+
"il-central-1": "780543022126",
223+
"ap-east-1": "871362719292",
224+
"ap-northeast-1": "763104351884",
225+
"ap-northeast-2": "763104351884",
226+
"ap-northeast-3": "364406365360",
227+
"ap-south-1": "763104351884",
228+
"ap-southeast-1": "763104351884",
229+
"ap-southeast-2": "763104351884",
230+
"ap-southeast-3": "907027046896",
231+
"ca-central-1": "763104351884",
232+
"cn-north-1": "727897471807",
233+
"cn-northwest-1": "727897471807",
234+
"eu-central-1": "763104351884",
235+
"eu-north-1": "763104351884",
236+
"eu-west-1": "763104351884",
237+
"eu-west-2": "763104351884",
238+
"eu-west-3": "763104351884",
239+
"eu-south-1": "692866216735",
240+
"me-south-1": "217643126080",
241+
"me-central-1": "914824155844",
242+
"sa-east-1": "763104351884",
243+
"us-east-1": "763104351884",
244+
"us-east-2": "763104351884",
245+
"us-gov-east-1": "446045086412",
246+
"us-gov-west-1": "442386744353",
247+
"us-west-1": "763104351884",
248+
"us-west-2": "763104351884",
249+
"ca-west-1": "204538143572",
250+
"eu-central-2": "380420809688",
251+
"ap-south-2": "772153158452",
252+
"eu-south-2": "503227376785",
253+
"ap-southeast-7": "590183813437",
254+
"ap-southeast-4": "457447274322",
255+
"ap-southeast-5": "550225433462",
256+
"mx-central-1": "637423239942"
257+
},
258+
"repository": "djl-inference",
259+
"tag_prefix": "0.31.0-lmi13.0.0-cu124"
260+
},
9261
"0.30.0": {
10262
"registries": {
11263
"af-south-1": "626614931356",

0 commit comments

Comments
 (0)