toml support for python 3.10 and below

lanl · Oct 31, 2024 · 1a48365 · 1a48365
1 parent dc634f6
commit 1a48365
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dsi/plugins/file_reader.py b/dsi/plugins/file_reader.py
@@ -6,7 +6,9 @@
 from pandas import DataFrame, read_csv, concat
 import re
 import yaml
-import tomllib
+try: import tomllib
+except ModuleNotFoundError: import pip._vendor.tomli as tomllib
+
 # import ast
 
 from dsi.plugins.metadata import StructuredMetadata