From 5f21d62d083ced6caeb29f5c0db5cdf7e8708080 Mon Sep 17 00:00:00 2001 From: Willy Douhard Date: Thu, 26 Sep 2024 15:00:15 +0200 Subject: [PATCH] fix: make sure each experiment run has its own id --- literalai/evaluation/experiment_item_run.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/literalai/evaluation/experiment_item_run.py b/literalai/evaluation/experiment_item_run.py index af6bcbf..b25fbe6 100644 --- a/literalai/evaluation/experiment_item_run.py +++ b/literalai/evaluation/experiment_item_run.py @@ -16,11 +16,8 @@ def __init__( self, client: "BaseLiteralClient", ): - self.id = str(uuid.uuid4()) + self.client = client EnvContextManager.__init__(self, client=client, env="experiment") - StepContextManager.__init__( - self, client=client, name="Experiment Run", type="run", id=self.id - ) def __call__(self, func): return experiment_item_run_decorator( @@ -30,7 +27,11 @@ def __call__(self, func): ) async def __aenter__(self): - active_experiment_item_run_id_var.set(self.id) + id = str(uuid.uuid4()) + StepContextManager.__init__( + self, client=self.client, name="Experiment Run", type="run", id=id + ) + active_experiment_item_run_id_var.set(id) await EnvContextManager.__aenter__(self) step = await StepContextManager.__aenter__(self) return step @@ -42,7 +43,11 @@ async def __aexit__(self, exc_type, exc_val, exc_tb): active_experiment_item_run_id_var.set(None) def __enter__(self): - active_experiment_item_run_id_var.set(self.id) + id = str(uuid.uuid4()) + StepContextManager.__init__( + self, client=self.client, name="Experiment Run", type="run", id=id + ) + active_experiment_item_run_id_var.set(id) EnvContextManager.__enter__(self) step = StepContextManager.__enter__(self) return step