From f2dc0bef55e330f80f790dd43a2dca679ad8ace0 Mon Sep 17 00:00:00 2001 From: Ram Rachum Date: Sat, 13 Apr 2024 10:21:20 +0300 Subject: [PATCH] Replace imp with importlib for Python 3.12 support --- setup.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 177fac3f..ced2e36f 100755 --- a/setup.py +++ b/setup.py @@ -15,10 +15,18 @@ # ============================================================================ """Install script for setuptools.""" -import imp +import sys +import importlib import setuptools +def load_source(module_name, path_to_file): + spec = importlib.util.spec_from_file_location(module_name, path_to_file) + module = importlib.util.module_from_spec(spec) + spec.loader.exec_module(module) + sys.modules[module_name] = module + return module + # Additional requirements for TensorFlow baselines, excluding OpenAI & Dopamine. # See baselines/README.md for more information. baselines_require = [ @@ -67,7 +75,7 @@ author='DeepMind', author_email='dm-bsuite-eng+os@google.com', license='Apache License, Version 2.0', - version=imp.load_source('_metadata', 'bsuite/_metadata.py').__version__, + version=load_source('_metadata', 'bsuite/_metadata.py').__version__, keywords='reinforcement-learning python machine-learning', packages=setuptools.find_packages(), install_requires=[