54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
import datetime
|
|
import time
|
|
|
|
|
|
def get_current_time_millis():
|
|
"""
|
|
Returns the time in milliseconds since the epoch as an integer number.
|
|
"""
|
|
return int(time.time() * 1000)
|
|
|
|
|
|
def conv_longdate_to_str(longdate, local_tz=True):
|
|
date_time = datetime.datetime.fromtimestamp(longdate / 1000.0)
|
|
str_long_date = date_time.strftime("%Y-%m-%d %H:%M:%S")
|
|
if local_tz:
|
|
if tzinfo := datetime.datetime.now().astimezone().tzinfo:
|
|
str_long_date += " " + tzinfo.tzname(date_time)
|
|
|
|
return str_long_date
|
|
|
|
|
|
class Timer:
|
|
"""
|
|
Measures elapsed time.
|
|
|
|
.. code-block:: python
|
|
|
|
from mlflow.utils.time import Timer
|
|
|
|
with Timer() as t:
|
|
...
|
|
|
|
print(f"Elapsed time: {t:.2f} seconds")
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.elapsed = 0.0
|
|
|
|
def __enter__(self):
|
|
self.elapsed = time.perf_counter()
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
self.elapsed = time.perf_counter() - self.elapsed
|
|
|
|
def __format__(self, format_spec: str) -> str:
|
|
return self.elapsed.__format__(format_spec)
|
|
|
|
def __repr__(self) -> str:
|
|
return self.elapsed.__repr__()
|
|
|
|
def __str__(self) -> str:
|
|
return self.elapsed.__str__()
|