Remove pkg_resources.resource_filename

mlflow/_doctor.py

@@ -2,7 +2,7 @@
 import platform
 
 import click
-import pkg_resources
+import pkg_resources  # noqa: TID251
 import yaml
 
 import mlflow

mlflow/models/container/__init__.py

@@ -10,10 +10,9 @@
 import shutil
 import signal
 import sys
+from pathlib import Path
 from subprocess import Popen, check_call
 
-from pkg_resources import resource_filename
-
 import mlflow
 import mlflow.version
 from mlflow import mleap, pyfunc
@@ -162,8 +161,8 @@ def _serve_pyfunc(model, env_manager):
         start_nginx = False
 
     if start_nginx:
-        nginx_conf = resource_filename(
-            mlflow.models.__name__, "container/scoring_server/nginx.conf"
+        nginx_conf = Path(mlflow.models.__file__).parent.joinpath(
+            "container", "scoring_server", "nginx.conf"
         )
 
         nginx = Popen(["nginx", "-c", nginx_conf]) if start_nginx else None

mlflow/utils/requirements_utils.py

@@ -17,7 +17,7 @@
 from typing import NamedTuple, Optional
 
 import importlib_metadata
-import pkg_resources
+import pkg_resources  # noqa: TID251
 from packaging.requirements import Requirement
 from packaging.version import InvalidVersion, Version
 
@@ -375,8 +375,7 @@ def _init_packages_to_modules_map():
 
 
 def _load_pypi_package_index():
-    pypi_index_path = pkg_resources.resource_filename(mlflow.__name__, "pypi_package_index.json")
-    with open(pypi_index_path) as f:
+    with Path(mlflow.__file__).parent.joinpath("pypi_package_index.json").open() as f:
         index_dict = json.load(f)
 
     return _PyPIPackageIndex(

pyproject.toml

@@ -50,6 +50,7 @@ select = [
   "SIM101",
   "T20",
   "TID252",
+  "TID251",
 ]
 force-exclude = true
 ignore = [
@@ -86,3 +87,6 @@ ban-relative-imports = "all"
 
 [tool.ruff.isort]
 forced-separate = ["tests"]
+
+[tool.ruff.flake8-tidy-imports.banned-api]
+"pkg_resources".msg = "We're migrating away from pkg_resources. Please use importlib.resources or importlib.metadata instead."