From 0a1dbe9d80aff304d81d4e2ce857ee0a10ba091f Mon Sep 17 00:00:00 2001
From: Christoph Alt <christoph.alt@fau.de>
Date: Wed, 5 Apr 2023 16:59:11 +0200
Subject: [PATCH] optionally use backport of importlib resources for json
import
---
plotting/roofline.py | 9 +++++++--
setup.py | 3 ++-
2 files changed, 9 insertions(+), 3 deletions(-)
diff --git a/plotting/roofline.py b/plotting/roofline.py
index ea007e4..8d5853f 100644
--- a/plotting/roofline.py
+++ b/plotting/roofline.py
@@ -1,7 +1,12 @@
import plotly.graph_objs as go
from dataclasses import dataclass
import json
-import importlib
+
+try:
+ import importlib.resources as pkg_resources
+except ImportError:
+ # Try backported to PY<37 `importlib_resources`.
+ import importlib_resources as pkg_resources
@dataclass
@@ -19,7 +24,7 @@ DEFAULT_MARKER = dict(size=10, color='green')
def get_data(file_name="roofline_data.json"):
- with importlib.resources.open_text(__package__, file_name) as file:
+ with pkg_resources.open_text(__package__, file_name) as file:
data = json.load(file)
return data
diff --git a/setup.py b/setup.py
index cb90c51..5103f7c 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,8 @@ setup(name="cb-util",
"requests",
"plotly",
"kaleido",
- "kadi-apy"
+ "kadi-apy",
+ "importlib_resources ; python_version<'3.7'",
],
setup_requires=['pytest-runner'],
tests_require=['pytest']
--
GitLab