From dea4f9d3a61e064fb9bd8d0abb5563a80515e688 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Grie=C3=9Fhaber?= <griesshaber@hdm-stuttgart.de>
Date: Fri, 25 Oct 2024 16:13:14 +0200
Subject: [PATCH] use corrext wandb entity
---
evoprompt/models.py | 6 +++---
main.py | 10 +++++++---
2 files changed, 10 insertions(+), 6 deletions(-)
diff --git a/evoprompt/models.py b/evoprompt/models.py
index cb7c4ff..8360c9e 100644
--- a/evoprompt/models.py
+++ b/evoprompt/models.py
@@ -1,7 +1,5 @@
-from collections.abc import Iterable
import hashlib
import inspect
-from itertools import zip_longest
import json
import logging
import random
@@ -10,14 +8,16 @@ import time
import warnings
from abc import ABC, abstractmethod
from argparse import ArgumentParser
+from collections.abc import Iterable
+from itertools import zip_longest
from pathlib import Path
from typing import Any, Callable, ClassVar
import llama_cpp
-from llama_cpp.llama_chat_format import format_llama3
import openai
import weave
from diskcache import Cache
+from llama_cpp.llama_chat_format import format_llama3
from evoprompt.cli import argument_parser
from evoprompt.opt_types import ModelUsage
diff --git a/main.py b/main.py
index ae7c518..66a16ac 100644
--- a/main.py
+++ b/main.py
@@ -2,11 +2,11 @@ import logging
import os
from typing import Any
-from dotenv import load_dotenv
-import wandb
import weave
+from dotenv import load_dotenv
from weave.trace.settings import UserSettings
+import wandb
from evoprompt.cli import argument_parser
from evoprompt.evolution import get_optimizer_class
from evoprompt.models import HfChat, Llama, LlamaChat, LLMModel
@@ -62,7 +62,11 @@ if __name__ == "__main__":
# TODO weave recently had 500 errors quite often, so we disable it for now
# weave_settings = UserSettings(disabled=False, print_call_link=False)
# weave.init(project_name=options.wandb_project, settings=weave_settings)
- wandb.init(project=options.wandb_project, config=options.__dict__)
+ wandb.init(
+ entity="prompt-optimization",
+ project=options.wandb_project,
+ config=options.__dict__,
+ )
# set up console logging and rnd
setup_console_logger(verbosity_level=options.verbose)
--
GitLab