chore: fix toml imports with extra requires

sudoblockio · Jan 8, 2024 · 634a488 · 634a488
1 parent 82f8f1f
commit 634a488
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/providers/toml/hooks/tomls.py b/providers/toml/hooks/tomls.py
@@ -1,7 +1,10 @@
 import os
 from typing import Union
 
-import tomli as toml
+try:
+    import tomllib as toml
+except ImportError:
+    import tomli as toml
 
 from tackle import BaseHook, Field
 

diff --git a/providers/toml/requirements.txt b/providers/toml/requirements.txt
@@ -0,0 +1,2 @@
+# this is temporary - py3.11 has this included
+tomli
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		# this is temporary - py3.11 has this included
		tomli