Skip to content

Commit

Permalink
Update env variables
Browse files Browse the repository at this point in the history
  • Loading branch information
adam-narozniak committed May 21, 2024
1 parent 7f5ad8e commit 535cbf2
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions datasets/flwr_datasets/telemetry.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,8 @@

from flwr.common.version import package_name, package_version

FLWR_DATASETS_TELEMETRY_ENABLED = os.getenv("FLWR_DATASETS_TELEMETRY_ENABLED", "1")
FLWR_DATASETS_TELEMETRY_LOGGING = os.getenv("FLWR_DATASETS_TELEMETRY_LOGGING", "0")
FLWR_TELEMETRY_ENABLED = os.getenv("FLWR_TELEMETRY_ENABLED", "1")
FLWR_TELEMETRY_LOGGING = os.getenv("FLWR_TELEMETRY_LOGGING", "0")

TELEMETRY_EVENTS_URL = "https://telemetry.flower.ai/api/v1/event"

Expand Down Expand Up @@ -193,12 +193,12 @@ def create_event(event_type: EventType, event_details: Optional[Dict[str, Any]])
"context": context,
}
payload_json = json.dumps(payload)
if FLWR_DATASETS_TELEMETRY_LOGGING == "1":
if FLWR_TELEMETRY_LOGGING == "1":
log(" - ".join([date, "POST", payload_json]))

# If telemetry is not disabled with setting FLWR_DATASETS_TELEMETRY_ENABLED=0
# If telemetry is not disabled with setting FLWR_TELEMETRY_ENABLED=0
# create a request and send it to the telemetry backend
if FLWR_DATASETS_TELEMETRY_ENABLED == "1":
if FLWR_TELEMETRY_ENABLED == "1":
request = urllib.request.Request(
url=TELEMETRY_EVENTS_URL,
data=payload_json.encode("utf-8"),
Expand All @@ -216,7 +216,7 @@ def create_event(event_type: EventType, event_details: Optional[Dict[str, Any]])

return response_json
except urllib.error.URLError as ex:
if FLWR_DATASETS_TELEMETRY_LOGGING == "1":
if FLWR_TELEMETRY_LOGGING == "1":
log(ex)

return "disabled"

0 comments on commit 535cbf2

Please sign in to comment.