From ab8491f1a3f759cc23fb6cc09d2469a2a1ad01ac Mon Sep 17 00:00:00 2001 From: Jonas Werner Date: Tue, 17 Feb 2026 10:52:34 +0900 Subject: [PATCH] fix(arena wand): updating wand notebook to fix install of dependencies and fix branding updating wand notebook to fix install of dependencies and fix branding --- .../notebooks/arena_wandb_model_train_evaluate_test.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/arena/notebooks/arena_wandb_model_train_evaluate_test.py b/arena/notebooks/arena_wandb_model_train_evaluate_test.py index 1c3264b..77843ab 100644 --- a/arena/notebooks/arena_wandb_model_train_evaluate_test.py +++ b/arena/notebooks/arena_wandb_model_train_evaluate_test.py @@ -36,12 +36,12 @@ def _(): @app.cell(hide_code=True) def _(mo): mo.md(r""" - # CoreWeave AI Labs: W&B Model Training Demo + # CoreWeave ARENA: W&B Model Training Demo /// admonition | About This Notebook type: info - This notebook is a demo adaptation of [W&B HuggingFace artifacts](https://colab.research.google.com/drive/1WObS8JQnVODKG-gxtWVokCvSkKjIWUjt?usp=sharing#scrollTo=06gXuaF8HTBD) to provide an example of AILabs use with W&B integration. + This notebook is a demo adaptation of [W&B HuggingFace artifacts](https://colab.research.google.com/drive/1WObS8JQnVODKG-gxtWVokCvSkKjIWUjt?usp=sharing#scrollTo=06gXuaF8HTBD) to provide an example of ARENA use with W&B integration. /// /// details | Pipeline Steps @@ -73,8 +73,10 @@ def _(mo): @app.cell def _(shell): - DEPENDENCIES = "wandb transformers datasets evaluate accelerate scikit-learn torch" - shell(f"pip install -q {DEPENDENCIES}") + import subprocess + import sys + DEPENDENCIES = ["wandb", "transformers", "datasets", "evaluate", "accelerate", "scikit-learn", "torch"] + subprocess.check_call([sys.executable, "-m", "pip", "install", "-q"] + DEPENDENCIES) return