post3

EpistasisLab · Sep 10, 2020 · 8eec4f9 · 8eec4f9
1 parent ebec5cc
commit 8eec4f9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/pmlb/_version.py b/pmlb/_version.py
@@ -23,4 +23,4 @@
 SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 """
 
-__version__ = '1.0.1.post2'
+__version__ = '1.0.1.post3'
diff --git a/pmlb/dataset_lists.py b/pmlb/dataset_lists.py
@@ -23,13 +23,10 @@
 SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 """
 import pathlib, pandas
-
-try: # used in repo
-    df_summary = pandas.read_csv('pmlb/all_summary_stats.tsv', sep='\t')
-except: # used in pypi pkg
-    import pkgutil
-    data = pkgutil.get_data('pmlb', 'all_summary_stats.tsv')
-    df_summary = pandas.read_csv(data , sep='\t')
+from io import StringIO
+import pkgutil
+data = pkgutil.get_data('pmlb', 'all_summary_stats.tsv')
+df_summary = pandas.read_csv(StringIO(data.decode("utf-8")) , sep='\t')
 regression_dataset_names = df_summary.query('task=="regression"')['dataset'].tolist()
 classification_dataset_names = df_summary.query('task=="classification"')['dataset'].tolist()
 dataset_names = regression_dataset_names + classification_dataset_names