drop guard for from ruamel.yaml import YAML

DanielYang59 · Oct 20, 2024 · 78f16e5 · 78f16e5
1 parent 0c9d844
commit 78f16e5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/src/monty/json.py b/src/monty/json.py
@@ -21,6 +21,8 @@
 from typing import Any
 from uuid import UUID, uuid4
 
+from ruamel.yaml import YAML
+
 try:
     import numpy as np
 except ImportError:
@@ -41,17 +43,11 @@
 except ImportError:
     bson = None
 
-try:
-    from ruamel.yaml import YAML
-except ImportError:
-    YAML = None
-
 try:
     import orjson
 except ImportError:
     orjson = None
 
-
 try:
     import torch
 except ImportError:

diff --git a/src/monty/serialization.py b/src/monty/serialization.py
@@ -9,10 +9,7 @@
 import os
 from typing import TYPE_CHECKING
 
-try:
-    from ruamel.yaml import YAML
-except ImportError:
-    YAML = None  # type: ignore[arg-type]
+from ruamel.yaml import YAML
 
 from monty.io import zopen
 from monty.json import MontyDecoder, MontyEncoder