Fix unresolvable toml import

ByteOtter · Aug 1, 2023 · 33140dc · 33140dc
1 parent 0a9a7e0
commit 33140dc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/nester/core/opinionated/opinionated_creation.py b/src/nester/core/opinionated/opinionated_creation.py
@@ -6,9 +6,13 @@
 import subprocess
 from typing import Any
 
-import toml
 import yaml
 
+try:
+    import tomllib as toml
+except ImportError:
+    import toml
+
 from . import exceptions, supported_linters