closes #1557

CamDavidsonPilon · Sep 13, 2023 · 003de09 · 003de09
1 parent c32ce59
commit 003de09
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lifelines/datasets/__init__.py b/lifelines/datasets/__init__.py
@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 import pandas as pd
-from pkg_resources import resource_filename
+from importlib import resources
 
 
 def _load_dataset(filename, **kwargs):
@@ -18,7 +18,7 @@ def _load_dataset(filename, **kwargs):
     -------
         output: DataFrame
     """
-    return pd.read_csv(resource_filename("lifelines", "datasets/" + filename), engine="python", **kwargs)
+    return pd.read_csv(resources.files("lifelines") / "datasets" / filename, engine="python", **kwargs)
 
 
 def load_recur(**kwargs):