Use register_model_metadata

stanford-crfm · Nov 1, 2023 · 56796cc · 56796cc
1 parent 7f6dc4e
commit 56796cc
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/helm/benchmark/model_deployment_registry.py b/src/helm/benchmark/model_deployment_registry.py
@@ -10,8 +10,7 @@
 from helm.common.object_spec import ObjectSpec
 from helm.benchmark.model_metadata_registry import (
     ModelMetadata,
-    ALL_MODELS_METADATA,
-    MODEL_NAME_TO_MODEL_METADATA,
+    register_model_metadata,
     get_model_metadata,
     TEXT_MODEL_TAG,
     FULL_FUNCTIONALITY_TEXT_MODEL_TAG,
@@ -124,8 +123,7 @@ def register_model_deployment(model_deployment: ModelDeployment) -> None:
             tags=[TEXT_MODEL_TAG, FULL_FUNCTIONALITY_TEXT_MODEL_TAG],
             deployment_names=[model_deployment.name],
         )
-        ALL_MODELS_METADATA.append(model_metadata)
-        MODEL_NAME_TO_MODEL_METADATA[model_name] = model_metadata
+        register_model_metadata(model_metadata)
         hlog(f"Registered default metadata for model {model_name}")