Skip to content

Commit e1f206b

Browse files
committed
format: black
1 parent eb5cecc commit e1f206b

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

src/sagemaker/local/entities.py

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -274,26 +274,26 @@ def describe(self):
274274
"AlgorithmSpecification": {
275275
"ContainerEntrypoint": self.container.container_entrypoint,
276276
},
277-
"FinalMetricDataList": self._extract_final_metrics()
277+
"FinalMetricDataList": self._extract_final_metrics(),
278278
}
279279
return response
280280

281281
def _extract_final_metrics(self):
282282
"""Extract metrics from container logs using metric definitions."""
283-
if not hasattr(self.container, 'logs') or not self.container.logs:
283+
if not hasattr(self.container, "logs") or not self.container.logs:
284284
return []
285285

286286
# Get metric definitions from container
287-
metric_definitions = getattr(self.container, 'metric_definitions', [])
287+
metric_definitions = getattr(self.container, "metric_definitions", [])
288288
if not metric_definitions:
289289
return []
290290

291291
final_metrics = []
292292
logs = self.container.logs
293293

294294
for metric_def in metric_definitions:
295-
metric_name = metric_def.get('Name')
296-
regex_pattern = metric_def.get('Regex')
295+
metric_name = metric_def.get("Name")
296+
regex_pattern = metric_def.get("Regex")
297297

298298
if not metric_name or not regex_pattern:
299299
continue
@@ -303,14 +303,17 @@ def _extract_final_metrics(self):
303303
if matches:
304304
# Use the last match as final metric
305305
final_value = float(matches[-1])
306-
final_metrics.append({
307-
'MetricName': metric_name,
308-
'Value': final_value,
309-
'Timestamp': self.end_time or datetime.now()
310-
})
306+
final_metrics.append(
307+
{
308+
"MetricName": metric_name,
309+
"Value": final_value,
310+
"Timestamp": self.end_time or datetime.now(),
311+
}
312+
)
311313

312314
return final_metrics
313315

316+
314317
class _LocalTransformJob(object):
315318
"""Placeholder docstring"""
316319

0 commit comments

Comments
 (0)