xax 0.3.15__tar.gz → 0.4.1__tar.gz

{xax-0.3.15/xax.egg-info → xax-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xax
-Version: 0.3.15
+Version: 0.4.1
 Summary: A library for fast Jax experimentation
 Home-page: https://github.com/kscalelabs/xax
 Author: Benjamin Bolte

{xax-0.3.15 → xax-0.4.1}/xax/__init__.py

@@ -12,7 +12,7 @@ and running the update script:
     python -m scripts.update_api --inplace
 """
 
-__version__ = "0.3.15"
+__version__ = "0.4.1"
 
 # This list shouldn't be modified by hand; instead, run the update script.
 __all__ = [

{xax-0.3.15 → xax-0.4.1}/xax/utils/experiments.py

@@ -5,6 +5,7 @@ import datetime
 import enum
 import functools
 import hashlib
+import importlib.metadata
 import inspect
 import itertools
 import json
@@ -29,7 +30,6 @@ from typing import Any, Iterator, Mapping, Self, Sequence, TypeVar, cast
 from urllib.parse import urlparse
 
 import git
-import pkg_resources
 import requests
 from jaxtyping import Array
 from omegaconf import MISSING, DictConfig, ListConfig, OmegaConf
@@ -464,7 +464,7 @@ def get_packages_with_versions() -> str:
     Returns:
         A dictionary of packages and their versions.
     """
-    packages = [(pkg.key, pkg.version) for pkg in pkg_resources.working_set]
+    packages = [(dist.metadata["Name"], dist.version) for dist in importlib.metadata.distributions()]
     return "\n".join([f"{key}=={version}" for key, version in sorted(packages)])
 
 

{xax-0.3.15 → xax-0.4.1/xax.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xax
-Version: 0.3.15
+Version: 0.4.1
 Summary: A library for fast Jax experimentation
 Home-page: https://github.com/kscalelabs/xax
 Author: Benjamin Bolte