From 3ff3e16ea80fb5218279795df5da537102e6d4eb Mon Sep 17 00:00:00 2001
From: Daniel Holanda <holand.daniel@gmail.com>
Date: Mon, 6 Jan 2025 10:59:10 -0800
Subject: [PATCH] Standardize Timestamps to Fixed Time Zone in TKML Runs (#257)

* Fix timezones when creating timestamps

* Lint
---
 setup.py                           | 1 +
 src/turnkeyml/sequence/sequence.py | 4 +++-
 src/turnkeyml/version.py           | 2 +-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index e96276e..7654314 100644
--- a/setup.py
+++ b/setup.py
@@ -45,6 +45,7 @@
         "GitPython>=3.1.40",
         "psutil",
         "wmi",
+        "pytz",
         # Conditional dependencies for ONNXRuntime backends
         "onnxruntime >=1.10.1;platform_system=='Linux' and extra != 'llm-oga-cuda'",
         "onnxruntime-directml >=1.19.0;platform_system=='Windows' and extra != 'llm-oga-cuda'",
diff --git a/src/turnkeyml/sequence/sequence.py b/src/turnkeyml/sequence/sequence.py
index dbf9a97..bd5bd63 100644
--- a/src/turnkeyml/sequence/sequence.py
+++ b/src/turnkeyml/sequence/sequence.py
@@ -4,6 +4,7 @@
 import copy
 from datetime import datetime
 from typing import List, Dict, Optional
+import pytz
 import turnkeyml.common.printing as printing
 import turnkeyml.common.exceptions as exp
 import turnkeyml.common.build as build
@@ -128,9 +129,10 @@ def launch(
         state.save_stat(fs.Keys.BUILD_STATUS, build.FunctionStatus.INCOMPLETE)
 
         # Save a timestamp so that we know the order of builds within a cache
+        pacific_tz = pytz.timezone("America/Los_Angeles")
         state.save_stat(
             fs.Keys.TIMESTAMP,
-            datetime.now(),
+            datetime.now(pacific_tz),
         )
 
         # Save the system information used for this build
diff --git a/src/turnkeyml/version.py b/src/turnkeyml/version.py
index 413c995..9de5176 100644
--- a/src/turnkeyml/version.py
+++ b/src/turnkeyml/version.py
@@ -1 +1 @@
-__version__ = "4.0.9"
+__version__ = "4.0.10"