nshtrainer 0.15.0__tar.gz → 0.15.1__tar.gz

{nshtrainer-0.15.0 → nshtrainer-0.15.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshtrainer
-Version: 0.15.0
+Version: 0.15.1
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshtrainer-0.15.0 → nshtrainer-0.15.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshtrainer"
-version = "0.15.0"
+version = "0.15.1"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"

{nshtrainer-0.15.0 → nshtrainer-0.15.1}/src/nshtrainer/loggers/wandb.py

@@ -1,9 +1,10 @@
+import importlib.metadata
 import logging
 from typing import TYPE_CHECKING, Literal
 
 import nshconfig as C
-import pkg_resources
 from lightning.pytorch import Callback, LightningModule, Trainer
+from packaging import version
 from typing_extensions import override
 
 from ..callbacks import WandbWatchConfig
@@ -120,9 +121,8 @@ class WandbLoggerConfig(CallbackConfigBase, BaseLoggerConfig):
                 import wandb  # type: ignore
 
                 # The minimum version that supports the new backend is 0.17.5
-                if pkg_resources.parse_version(
-                    wandb.__version__
-                ) < pkg_resources.parse_version("0.17.5"):
+                wandb_version = version.parse(importlib.metadata.version("wandb"))
+                if wandb_version < version.parse("0.17.5"):
                     log.warning(
                         "The version of WandB installed does not support the `wandb-core` backend "
                         f"(expected version >= 0.17.5, found version {wandb.__version__}). "

{nshtrainer-0.15.0 → nshtrainer-0.15.1}/src/nshtrainer/scripts/find_packages.py

@@ -40,7 +40,6 @@ def main():
 
     if args.exclude_std:
         std_libs = set(sys.stdlib_module_names)
-        std_libs.update({"pkg_resources"})
         all_imports = all_imports - std_libs
 
     for package in sorted(all_imports):