Commit d456c369 authored by jay's avatar jay
Browse files

Fixes redis push to work queue.

parent 3be59da9
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ import time
import numpy as np

from plurmy import slurm_walltime_to_seconds
from autocnet.utils.serializers import object_hook
from autocnet.utils.serializers import JsonEncoder, object_hook

def pop_computetime_push(queue, inqueue, outqueue):
    """
@@ -32,7 +32,7 @@ def pop_computetime_push(queue, inqueue, outqueue):
    msg['max_time'] = time.time() + slurm_walltime_to_seconds(msg['walltime'])

    # Push the message to the processing queue with the updated max_time
    queue.rpush(outqueue, json.dumps(msg))
    queue.rpush(outqueue, json.dumps(msg, cls=JsonEncoder))

    return msg