Skip to content

Commit

Permalink
Fixing docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
ghyadav committed Dec 27, 2024
1 parent 30a275d commit 89495bf
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@


class TelemetryConstants:
"""Telemetry Constants."""

APP_INSIGHT_HANDLER_NAME = "AppInsightsHandler"
NON_PII_MESSAGE = '[Hidden as it may contain PII]'

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@

"""Logging utilities for the evaluator."""


from azure.ml.component.run import CoreRun
import platform

Expand Down Expand Up @@ -235,6 +234,7 @@ def format(self, record: logging.LogRecord) -> str:

class CustomLogRecord(logging.LogRecord):
"""Custom Log Record class for App Insights."""

def __init__(self, *args, **kwargs):
"""__init__."""
super().__init__(*args, **kwargs)
Expand All @@ -243,7 +243,7 @@ def __init__(self, *args, **kwargs):

# Step 2: Set the custom LogRecord factory
def custom_log_record_factory(*args, **kwargs):
"""Custom Log Record Factory for App Insights."""
"""Get CustomLogRecord for App Insights."""
return CustomLogRecord(*args, **kwargs)


Expand Down

0 comments on commit 89495bf

Please sign in to comment.