From 06ee2791d052cc48f175b961959c29a42879ad5a Mon Sep 17 00:00:00 2001 From: Felipe Mello Date: Wed, 3 Dec 2025 10:37:40 -0800 Subject: [PATCH] remove pytz --- src/forge/observability/metrics.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/forge/observability/metrics.py b/src/forge/observability/metrics.py index 34b0839df..2c5a4663f 100644 --- a/src/forge/observability/metrics.py +++ b/src/forge/observability/metrics.py @@ -10,14 +10,12 @@ import json import logging import os +import time from abc import ABC, abstractmethod from dataclasses import dataclass -from datetime import datetime from enum import Enum from typing import Any, Dict, List -import pytz - from forge.observability.utils import get_proc_name_with_rank from forge.util.logging import get_logger, log_once @@ -91,7 +89,7 @@ def accumulator_class(self): class Metric: """Container for metric data including key, value, reduction type, and timestamp. - Timestamp is automatically set to current UTC time if not provided. + Timestamp is automatically set to current time if not provided. """ key: str @@ -101,8 +99,7 @@ class Metric: def __post_init__(self): if self.timestamp is None: - # Always record in UTC timezone - self.timestamp = datetime.now(pytz.UTC).timestamp() + self.timestamp = time.time() def record_metric(key: str, value: Any, reduction: Reduce = Reduce.MEAN) -> None: