Merge pull request #813 from petiay/pkg_resources_replace

replace pkg_resources with importlib_resources

beast/physicsmodel/stars/ezmist/mist.py

@@ -8,7 +8,7 @@
 :author: MF
 """
 
-from pkg_resources import resource_filename
+from importlib_resources import files
 
 from urllib import request
 from urllib.request import urlopen
@@ -20,12 +20,11 @@
 
 py3k = True
 
-localpath = resource_filename("beast", "physicsmodel/stars/ezmist")
+localpath = str(files('beast.physicsmodel.stars.ezmist').joinpath('mist.json'))
 
-with open(localpath + "/mist.json") as f:
+with open(localpath) as f:
     _cfg = json.load(f)
 
-
 # Help messages
 # -------------
 

beast/physicsmodel/stars/ezpadova/parsec.py

@@ -6,7 +6,7 @@
 :author: MF
 """
 
-from pkg_resources import resource_filename
+from importlib_resources import files
 
 from urllib.parse import urlencode
 from urllib import request
@@ -21,9 +21,9 @@
 
 py3k = True
 
-localpath = resource_filename("beast", "physicsmodel/stars/ezpadova")
+localpath = str(files('beast.physicsmodel.stars.ezpadova').joinpath('parsec.json'))
 
-with open(localpath + "/parsec.json") as f:
+with open(localpath) as f:
     _cfg = json.load(f)
     map_carbon_stars = _cfg["map_carbon_stars"]
     map_phot = _cfg["map_phot"]

beast/tools/compare_spec_type.py

@@ -1,14 +1,16 @@
 import numpy as np
 from collections import defaultdict
 from scipy.interpolate import RegularGridInterpolator
-import pkg_resources
+from importlib_resources import files
 
 from astropy.table import Table
 from astropy.coordinates import SkyCoord
 from astropy import units as u
 
 import beast.plotting.plot_compare_spec_type as plot_match
 
+__all__ = ["compare_spec_type"]
+
 
 def compare_spec_type(
     phot_cat_file,
@@ -258,9 +260,8 @@ def setup_teff_table():
         function to interpolate across the T_eff table
     """
 
-    # read in the table
-    data_path = pkg_resources.resource_filename("beast", "tools/data/")
-    teff_table = Table.read(data_path + "effective_temperature.txt", format="ascii")
+    data_path = str(files("beast.tools.data").joinpath("effective_temperature.txt"))
+    teff_table = Table.read(data_path, format="ascii")
     # make each row a number rather than a letter+number
     teff_table["row_id"] = np.zeros(len(teff_table))
     for i in range(len(teff_table)):
@@ -299,9 +300,8 @@ def setup_logg_table():
         function to interpolate across the log(g) table
     """
 
-    # read in the table
-    data_path = pkg_resources.resource_filename("beast", "tools/data/")
-    logg_table = Table.read(data_path + "surface_gravity.txt", format="ascii")
+    data_path = str(files("beast.tools.data").joinpath("surface_gravity.txt"))
+    logg_table = Table.read(data_path, format="ascii")
     # make each row a number rather than a letter+number
     logg_table["row_id"] = np.zeros(len(logg_table))
     for i in range(len(logg_table)):

setup.cfg

@@ -31,6 +31,7 @@ install_requires =
     asdf
     stsynphot
     pandeia.engine
+    importlib_resources
 
 [options.entry_points]
 console_scripts =