Paul Gavrikov commited on
Commit
ca10c01
·
1 Parent(s): dbd32a8

fix persistent rate limits

Browse files
Files changed (1) hide show
  1. rate_limiter.py +6 -5
rate_limiter.py CHANGED
@@ -38,7 +38,7 @@ class RateLimitState:
38
  "daily_count": self.daily_count,
39
  "total_count": self.total_count,
40
  "last_access_time": self.last_access_time,
41
- "last_access_date": str(self.last_access_date),
42
  }
43
 
44
  @staticmethod
@@ -47,14 +47,15 @@ class RateLimitState:
47
  state.daily_count = data.get("daily_count", 0)
48
  state.total_count = data.get("total_count", 0)
49
  state.last_access_time = data.get("last_access_time")
50
- state.last_access_date = datetime.datetime.strptime(
51
- data.get("last_access_date"), "%Y-%m-%d"
52
- ).date()
53
  return state
 
 
 
54
 
55
 
56
  class RateLimiter:
57
- def __init__(self, config: RateLimitConfig, flush_interval=60):
58
  """
59
  flush_interval = how often (seconds) to push logs to HF dataset
60
  """
 
38
  "daily_count": self.daily_count,
39
  "total_count": self.total_count,
40
  "last_access_time": self.last_access_time,
41
+ "last_access_date": self.last_access_date.strftime("%Y-%m-%d"),
42
  }
43
 
44
  @staticmethod
 
47
  state.daily_count = data.get("daily_count", 0)
48
  state.total_count = data.get("total_count", 0)
49
  state.last_access_time = data.get("last_access_time")
50
+ state.last_access_date = data.get("last_access_date").date()
 
 
51
  return state
52
+
53
+ def __repr__(self):
54
+ return json.dumps(self.to_dict(), default=_json_dumper)
55
 
56
 
57
  class RateLimiter:
58
+ def __init__(self, config: RateLimitConfig, flush_interval=30):
59
  """
60
  flush_interval = how often (seconds) to push logs to HF dataset
61
  """