migrate all yaml loaders to ruamel.yaml

common/sampling.py

@@ -3,7 +3,7 @@
 import aiofiles
 import json
 import pathlib
-import yaml
+from ruamel.yaml import YAML
 from copy import deepcopy
 from loguru import logger
 from pydantic import AliasChoices, BaseModel, Field
@@ -11,6 +11,8 @@ from typing import Dict, List, Optional, Union
 
 from common.utils import unwrap, prune_dict
 
+yaml = YAML()
+
 
 # Common class for sampler params
 class BaseSamplerRequest(BaseModel):