From 8ce6960bea08dc60e7727bdb11c817545378d0bb Mon Sep 17 00:00:00 2001 From: WyrmyCrawl Date: Tue, 29 Sep 2026 12:41:53 +0530 Subject: [PATCH] Updated deprecated pkg_resources dependency --- src/deimkit/config.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/src/deimkit/config.py b/src/deimkit/config.py index 40d0334e..bda25e22 100644 --- a/src/deimkit/config.py +++ b/src/deimkit/config.py @@ -1,7 +1,7 @@ import os import pprint -import pkg_resources +from importlib import resources as _resources from .engine.core import BaseConfig, YAMLConfig @@ -54,12 +54,16 @@ def from_model_name(cls, model_name, **kwargs): # Try to find the config file in several possible locations possible_paths = [ # Option 1: Look in the package's installed configs directory - pkg_resources.resource_filename( - "deimkit", f"configs/{model_name}_coco.yml" - ), - pkg_resources.resource_filename( - "deimkit", f"configs/deim_dfine/{model_name}_coco.yml" - ), + os.fspath( + _resources.files("deimkit").joinpath( + f"configs/{model_name}_coco.yml" + ) + ), + os.fspath( + _resources.files("deimkit").joinpath( + f"configs/deim_dfine/{model_name}_coco.yml" + ) + ), # Option 2: Look relative to the current working directory # os.path.join(os.getcwd(), f'configs/{model_name}_coco.yml'), # os.path.join(os.getcwd(), f'configs/deim_dfine/{model_name}_coco.yml'),