From 0fd39d062a4b6a7ee3911a480ea721f4a8b70293 Mon Sep 17 00:00:00 2001 From: Joschka Braun Date: Thu, 6 Jun 2024 17:46:00 -0400 Subject: [PATCH] fix: experiment run name creation --- parea/experiment/experiment.py | 8 ++++++-- pyproject.toml | 2 +- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/parea/experiment/experiment.py b/parea/experiment/experiment.py index 1fd3f942..4bd41d97 100644 --- a/parea/experiment/experiment.py +++ b/parea/experiment/experiment.py @@ -237,7 +237,9 @@ def run(self, run_name: Optional[str] = None) -> None: try: self._gen_run_name_if_none(run_name) - experiment_schema: ExperimentSchema = self.p.create_experiment(CreateExperimentRequest(name=self.experiment_name, run_name=run_name, metadata=self.metadata)) + experiment_schema: ExperimentSchema = self.p.create_experiment( + CreateExperimentRequest(name=self.experiment_name, run_name=self.run_name, metadata=self.metadata) + ) self.experiment_uuid = experiment_schema.uuid self.experiment_stats = asyncio.run( experiment(self.experiment_name, self.run_name, self.data, self.func, self.p, self.experiment_uuid, self.n_trials, self.dataset_level_evals, self.n_workers) @@ -259,7 +261,9 @@ async def arun(self, run_name: Optional[str] = None) -> None: try: self._gen_run_name_if_none(run_name) - experiment_schema: ExperimentSchema = await self.p.acreate_experiment(CreateExperimentRequest(name=self.experiment_name, run_name=run_name, metadata=self.metadata)) + experiment_schema: ExperimentSchema = await self.p.acreate_experiment( + CreateExperimentRequest(name=self.experiment_name, run_name=self.run_name, metadata=self.metadata) + ) self.experiment_uuid = experiment_schema.uuid self.experiment_stats = await experiment( self.experiment_name, self.run_name, self.data, self.func, self.p, self.experiment_uuid, self.n_trials, self.dataset_level_evals, self.n_workers diff --git a/pyproject.toml b/pyproject.toml index c8cff5fe..c3bb6f96 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api" [tool.poetry] name = "parea-ai" packages = [{ include = "parea" }] -version = "0.2.169" +version = "0.2.170" description = "Parea python sdk" readme = "README.md" authors = ["joel-parea-ai "]