@@ -202,7 +202,7 @@ def version_experiment(
202202
203203 return commit_hash
204204
205- # %% ../../nbs/project/experiments.ipynb 22
205+ # %% ../../nbs/project/experiments.ipynb 21
206206def cleanup_experiment_branches (
207207 prefix : str = "ragas/" ,
208208 repo_path : t .Union [str , Path , None ] = None ,
@@ -281,26 +281,26 @@ def cleanup_experiment_branches(
281281
282282 return deleted_branches
283283
284- # %% ../../nbs/project/experiments.ipynb 25
284+ # %% ../../nbs/project/experiments.ipynb 24
285285@t .runtime_checkable
286286class ExperimentProtocol (t .Protocol ):
287287 async def __call__ (self , * args , ** kwargs ): ...
288288 async def run_async (self , name : str , dataset : Dataset ): ...
289289
290- # %% ../../nbs/project/experiments.ipynb 26
290+ # %% ../../nbs/project/experiments.ipynb 25
291291from .naming import MemorableNames
292292
293- # %% ../../nbs/project/experiments.ipynb 27
293+ # %% ../../nbs/project/experiments.ipynb 26
294294memorable_names = MemorableNames ()
295295
296- # %% ../../nbs/project/experiments.ipynb 28
296+ # %% ../../nbs/project/experiments.ipynb 27
297297@patch
298298def experiment (
299299 self : Project ,
300300 experiment_model ,
301301 name_prefix : str = "" ,
302302 save_to_git : bool = True ,
303- stage_all : bool = False ,
303+ stage_all : bool = True ,
304304):
305305 """Decorator for creating experiment functions without Langfuse integration.
306306
@@ -400,11 +400,11 @@ async def run_async(
400400
401401 return decorator
402402
403- # %% ../../nbs/project/experiments.ipynb 32
403+ # %% ../../nbs/project/experiments.ipynb 31
404404# this one we have to clean up
405405from langfuse .decorators import observe
406406
407- # %% ../../nbs/project/experiments.ipynb 33
407+ # %% ../../nbs/project/experiments.ipynb 32
408408@patch
409409def langfuse_experiment (self : Project , experiment_model , name_prefix : str = "" ):
410410 """Decorator for creating experiment functions with Langfuse integration.
@@ -436,7 +436,7 @@ async def langfuse_wrapped_func(*args, **kwargs):
436436
437437 return decorator
438438
439- # %% ../../nbs/project/experiments.ipynb 40
439+ # %% ../../nbs/project/experiments.ipynb 39
440440from mlflow import trace
441441
442442
@@ -481,7 +481,7 @@ async def run_async_with_mlflow(dataset: Dataset, name: t.Optional[str] = None):
481481
482482 return decorator
483483
484- # %% ../../nbs/project/experiments.ipynb 41
484+ # %% ../../nbs/project/experiments.ipynb 40
485485import logging
486486from ..utils import plot_experiments_as_subplots
487487
0 commit comments