From 305fdc41fbefeca68901879b5545dfdd12047668 Mon Sep 17 00:00:00 2001 From: jebus Date: Wed, 12 Nov 2025 11:04:00 +1300 Subject: [PATCH] [Fix] Replace datetime.now() with Djangos timezone.now() to get rid of NaiveTimestamp warning (#191) Co-authored-by: Tim Lorsbach Reviewed-on: https://git.envipath.com/enviPath/enviPy/pulls/191 --- epdb/tasks.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/epdb/tasks.py b/epdb/tasks.py index b872d4a9..5bf46e38 100644 --- a/epdb/tasks.py +++ b/epdb/tasks.py @@ -1,15 +1,15 @@ import csv import io import logging -from datetime import datetime from typing import Any, Callable, List, Optional from uuid import uuid4 from celery import shared_task from celery.utils.functional import LRUCache +from django.utils import timezone from epdb.logic import SPathway -from epdb.models import EPModel, JobLog, Node, Package, Pathway, Rule, Setting, User, Edge +from epdb.models import Edge, EPModel, JobLog, Node, Package, Pathway, Rule, Setting, User logger = logging.getLogger(__name__) ML_CACHE = LRUCache(3) # Cache the three most recent ML models to reduce load times. @@ -29,7 +29,7 @@ def dispatch_eager(user: "User", job: Callable, *args, **kwargs): log.task_id = uuid4() log.job_name = job.__name__ log.status = "SUCCESS" - log.done_at = datetime.now() + log.done_at = timezone.now() log.task_result = str(x) if x else None log.save()