Lazily load yaml to avoid libyaml dependency

oughtinc · Jul 3, 2023 · 59e1c36 · 59e1c36
1 parent fa548bb
commit 59e1c36
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ice/metrics/gold_standards.py b/ice/metrics/gold_standards.py
@@ -16,8 +16,6 @@
 from pydantic import BaseModel
 from pydantic.generics import GenericModel
 from structlog.stdlib import get_logger
-from yaml import CLoader as Loader
-from yaml import load
 
 from ice.paper import Paper
 from ice.settings import settings
@@ -60,6 +58,9 @@ class Config:
 def _parse_answer(
     _answer: str, model: Type[ParsedGoldStandardType]
 ) -> ParsedGoldStandardType:
+    from yaml import CLoader as Loader
+    from yaml import load
+
     data = load(_answer, Loader=Loader)
     return model.parse_obj(data)