core[patch]: Support .yml extension for YAML (#16783)

- **Description:**

[AS-IS] When dealing with a yaml file, the extension must be .yaml.  

[TO-BE] In the absence of extension length constraints in the OS, the
extension of the YAML file is yaml, but control over the yml extension
must still be made.

It's as if it's an error because it's a .jpg extension in jpeg support.

  - **Issue:** - 

  - **Dependencies:**
no dependencies required for this change,
This commit is contained in:
Theo / Taeyoon Kang
2024-02-13 12:57:20 +09:00
committed by GitHub
parent cd00a87db7
commit 1987f905ed
9 changed files with 11 additions and 11 deletions

View File

@@ -35,7 +35,7 @@ def load_llm(file: Union[str, Path]) -> BaseLLM:
if file_path.suffix == ".json":
with open(file_path) as f:
config = json.load(f)
elif file_path.suffix == ".yaml":
elif file_path.suffix.endswith((".yaml", ".yml")):
with open(file_path, "r") as f:
config = yaml.safe_load(f)
else: