Skip to content

Commit 605b13a

Browse files
authored
Make backward compatible (#34607)
1 parent a545697 commit 605b13a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

sdk/ml/azure-ai-ml/azure/ai/ml/_telemetry/activity.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,9 +283,12 @@ def wrapper(*args, **kwargs):
283283
logger.package_logger, activity_name or f.__name__, activity_type, custom_dimensions
284284
):
285285
return f(*args, **kwargs)
286-
else:
286+
elif hasattr(logger, "package_logger"):
287287
with log_activity(logger.package_logger, activity_name or f.__name__, activity_type, custom_dimensions):
288288
return f(*args, **kwargs)
289+
else:
290+
with log_activity(logger, activity_name or f.__name__, activity_type, custom_dimensions):
291+
return f(*args, **kwargs)
289292

290293
return wrapper
291294

0 commit comments

Comments
 (0)