Skip to content

Commit

Permalink
Rename TELEM_LABELS to TELEMETRY_LABELS for consistency
Browse files Browse the repository at this point in the history
  • Loading branch information
AdrianSosic committed Feb 19, 2025
1 parent f1721bf commit 7917af8
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 13 deletions.
8 changes: 4 additions & 4 deletions baybe/campaign.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@
from baybe.surrogates.base import SurrogateProtocol
from baybe.targets.base import Target
from baybe.telemetry import (
TELEM_LABELS,
TELEMETRY_LABELS,
telemetry_record_recommended_measurement_percentage,
telemetry_record_value,
)
Expand Down Expand Up @@ -303,7 +303,7 @@ def add_measurements(
self._searchspace_metadata.loc[idxs_matched, _MEASURED] = True

# Telemetry
telemetry_record_value(TELEM_LABELS["COUNT_ADD_RESULTS"], 1)
telemetry_record_value(TELEMETRY_LABELS["COUNT_ADD_RESULTS"], 1)
telemetry_record_recommended_measurement_percentage(
self._cached_recommendation, data, self.parameters
)
Expand Down Expand Up @@ -501,8 +501,8 @@ def recommend(
self._searchspace_metadata.loc[rec.index, _RECOMMENDED] = True

# Telemetry
telemetry_record_value(TELEM_LABELS["COUNT_RECOMMEND"], 1)
telemetry_record_value(TELEM_LABELS["BATCH_SIZE"], batch_size)
telemetry_record_value(TELEMETRY_LABELS["COUNT_RECOMMEND"], 1)
telemetry_record_value(TELEMETRY_LABELS["BATCH_SIZE"], batch_size)

return rec

Expand Down
8 changes: 4 additions & 4 deletions baybe/searchspace/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@
)
from baybe.searchspace.validation import validate_parameters
from baybe.serialization import SerialMixin, converter, select_constructor_hook
from baybe.telemetry import TELEM_LABELS, telemetry_record_value
from baybe.telemetry import TELEMETRY_LABELS, telemetry_record_value
from baybe.utils.plotting import to_string


Expand Down Expand Up @@ -85,10 +85,10 @@ def __attrs_post_init__(self):
validate_constraints(self.constraints, self.parameters)

# Telemetry
telemetry_record_value(TELEM_LABELS["COUNT_SEARCHSPACE_CREATION"], 1)
telemetry_record_value(TELEM_LABELS["NUM_PARAMETERS"], len(self.parameters))
telemetry_record_value(TELEMETRY_LABELS["COUNT_SEARCHSPACE_CREATION"], 1)
telemetry_record_value(TELEMETRY_LABELS["NUM_PARAMETERS"], len(self.parameters))
telemetry_record_value(
TELEM_LABELS["NUM_CONSTRAINTS"],
TELEMETRY_LABELS["NUM_CONSTRAINTS"],
len(self.constraints) if self.constraints else 0,
)

Expand Down
4 changes: 2 additions & 2 deletions baybe/telemetry/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"""

from baybe.telemetry.api import (
TELEM_LABELS,
TELEMETRY_LABELS,
VARNAME_TELEMETRY_ENABLED,
VARNAME_TELEMETRY_HOSTNAME,
VARNAME_TELEMETRY_USERNAME,
Expand All @@ -13,7 +13,7 @@
)

__all__ = [
"TELEM_LABELS",
"TELEMETRY_LABELS",
"VARNAME_TELEMETRY_ENABLED",
"VARNAME_TELEMETRY_HOSTNAME",
"VARNAME_TELEMETRY_USERNAME",
Expand Down
6 changes: 3 additions & 3 deletions baybe/telemetry/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from baybe.utils.dataframe import fuzzy_row_match

# Telemetry labels for metrics
TELEM_LABELS = {
TELEMETRY_LABELS = {
"RECOMMENDED_MEASUREMENTS_PERCENTAGE": "value_recommended-measurements-percentage",
"BATCH_SIZE": "value_batch-size",
"COUNT_ADD_RESULTS": "count_add-results",
Expand Down Expand Up @@ -84,7 +84,7 @@ def telemetry_record_recommended_measurement_percentage(
from baybe.telemetry._telemetry import transmission_queue

if cached_recommendation.empty:
transmission_queue.put((TELEM_LABELS["NAKED_INITIAL_MEASUREMENTS"], 1))
transmission_queue.put((TELEMETRY_LABELS["NAKED_INITIAL_MEASUREMENTS"], 1))
else:
recommended_measurements_percentage = (
len(fuzzy_row_match(cached_recommendation, measurements, parameters))
Expand All @@ -93,7 +93,7 @@ def telemetry_record_recommended_measurement_percentage(
)
transmission_queue.put(
(
TELEM_LABELS["RECOMMENDED_MEASUREMENTS_PERCENTAGE"],
TELEMETRY_LABELS["RECOMMENDED_MEASUREMENTS_PERCENTAGE"],
recommended_measurements_percentage,
)
)

0 comments on commit 7917af8

Please sign in to comment.