Replace modellogger with airrlogger in codebase and update dependencies in pyproject.toml

Author: superdosh

pyproject.toml

@@ -82,7 +82,7 @@ dependencies = [
     "pyarrow>=22.0.0,<24.0.0",
     "tiktoken>=0.12.0,<1.0.0",
     "uvicorn>=0.38,<0.41",
-    "modellogger @ git+https://github.com/mlcommons/modellogger.git",
+    "airrlogger",
 ]
 
 [project.urls]

src/modelbench/benchmark_runner.py

@@ -8,7 +8,7 @@
 from datetime import datetime
 from typing import Any, Iterable, Optional, Sequence
 
-from modellogger.log_config import get_logger
+from airrlogger.log_config import get_logger
 from pydantic import BaseModel
 from tqdm import tqdm
 

src/modelbench/cli.py

@@ -16,17 +16,17 @@
 
 import click
 import termcolor
+from airrlogger.log_config import configure_logging
 from click import echo
-from modellogger.log_config import configure_logging
 from rich.console import Console
 from rich.table import Table
 
 import modelgauge.annotators.cheval.registration  # noqa: F401
 from modelbench.benchmark_runner import BenchmarkRun, BenchmarkRunner, JsonRunTracker, TqdmRunTracker
 from modelbench.benchmarks import GeneralPurposeAiChatBenchmarkV1, SecurityBenchmark
 from modelbench.consistency_checker import (
-    ConsistencyCheckError,
     ConsistencyChecker,
+    ConsistencyCheckError,
     summarize_consistency_check_results,
 )
 from modelbench.record import dump_json

src/modelgauge/annotation_pipeline.py

@@ -1,6 +1,6 @@
 import time
 
-from modellogger.log_config import get_logger
+from airrlogger.log_config import get_logger
 from pydantic import BaseModel
 
 from modelgauge.annotator import Annotator

src/modelgauge/annotators/llama_guard_annotator.py

@@ -3,7 +3,7 @@
 from string import Template
 from typing import Callable, Dict, List, Optional
 
-from modellogger.log_config import get_logger
+from airrlogger.log_config import get_logger
 
 from modelgauge.annotation import SafetyAnnotation
 from modelgauge.annotator import Annotator

src/modelgauge/cli.py

@@ -4,7 +4,7 @@
 from typing import Optional, Sequence, Type
 
 import click
-from modellogger.log_config import get_logger
+from airrlogger.log_config import get_logger
 
 from modelgauge.annotator import Annotator
 from modelgauge.annotator_registry import ANNOTATORS

src/modelgauge/command_line.py

@@ -3,7 +3,7 @@
 import sys
 
 import click
-from modellogger.log_config import configure_logging
+from airrlogger.log_config import configure_logging
 
 from modelgauge.annotator_registry import ANNOTATORS
 from modelgauge.cli_lazy import LOAD_ALL, LazyModuleImportGroup

src/modelgauge/config.py

@@ -5,7 +5,7 @@
 from typing import Dict, Mapping, Sequence
 
 import tomli
-from modellogger.log_config import get_logger
+from airrlogger.log_config import get_logger
 
 from modelgauge import config_templates
 from modelgauge.secret_values import MissingSecretValues, RawSecrets, SecretDescription

src/modelgauge/general.py

@@ -7,7 +7,7 @@
 import time
 from typing import List, Optional, Set, Type, TypeVar
 
-from modellogger.log_config import get_logger
+from airrlogger.log_config import get_logger
 from tqdm import tqdm
 
 # Type vars helpful in defining templates.

src/modelgauge/pipeline_runner.py

@@ -3,7 +3,7 @@
 from abc import ABC, abstractmethod
 from multiprocessing.pool import ThreadPool
 
-from modellogger.log_config import get_logger
+from airrlogger.log_config import get_logger
 
 from modelgauge.annotation_pipeline import AnnotatorAssigner, AnnotatorSink, AnnotatorSource, AnnotatorWorkers
 from modelgauge.dataset import AnnotationDataset, PromptDataset, PromptResponseDataset