Skip to content

Commit dcac1e2

Browse files
Merge pull request #2134 from FedML-AI/raphael/fix-timezone-autoscale
[Deploy] Fix timezone issue when using pandas
2 parents a24e962 + e8844d3 commit dcac1e2

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

  • python/fedml/computing/scheduler/model_scheduler/autoscaler

python/fedml/computing/scheduler/model_scheduler/autoscaler/autoscaler.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,11 +50,11 @@ def filter_by_timestamp(cls,
5050
filtered = metrics
5151
if before_now_minutes:
5252
less_than_ts = \
53-
str(pd.Timestamp.now() - pd.Timedelta(minutes=before_now_minutes))
53+
str(pd.Timestamp.utcnow().replace(tzinfo=None) - pd.Timedelta(minutes=before_now_minutes))
5454
filtered = metrics.query("'{}' <= {}".format(less_than_ts, "timestamp"))
5555
if before_now_seconds:
5656
less_than_ts = \
57-
str(pd.Timestamp.now() - pd.Timedelta(seconds=before_now_seconds))
57+
str(pd.Timestamp.utcnow().replace(tzinfo=None) - pd.Timedelta(seconds=before_now_seconds))
5858
filtered = metrics.query("'{}' <= {}".format(less_than_ts, "timestamp"))
5959
return filtered
6060

@@ -151,6 +151,7 @@ def scale_operation_query_concurrency(cls,
151151

152152
# Otherwise, we proceed as normal.
153153
queries_num = period_data.shape[0]
154+
logging.info(f"Detect {queries_num} of requests in {concurrent_query_policy.window_size_secs} seconds")
154155

155156
try:
156157
# QSR: Queries per Second per Replica: (Number of Queries / Number of Current Replicas) / Window Size

0 commit comments

Comments
 (0)