Updated pyproject.toml

Removed unused testing deps:
- ale-py, gymnasium

Replaced default torch installation with cpu version only

Author: vfdev-5

pyproject.toml

@@ -40,7 +40,6 @@ testing = [
     "clu",
     "clu<=0.0.9; python_version<'3.10'",
     "einops",
-    "gymnasium[atari]",
     "jaxlib",
     "jaxtyping",
     "jraph>=0.0.6dev0",
@@ -61,11 +60,10 @@ testing = [
     "tensorflow_text>=2.11.0; platform_system!='Darwin' and python_version < '3.13'",
     "tensorflow_datasets",
     "tensorflow>=2.12.0; python_version<'3.13'", # to fix Numpy np.bool8 deprecation error
-    "tensorflow>=2.20.0; python_version>='3.13'",
+    "tensorflow>=2.20.0; python_version>='3.13' and python_version<'3.14'",
     "torch",
     "treescope>=0.1.1; python_version>='3.10'",
     "cloudpickle>=3.0.0",
-    "ale-py>=0.10.2",
 ]
 docs = [
     "sphinx==6.2.1",
@@ -237,3 +235,11 @@ quote-style = "single"
 [tool.uv]
 # Ignore uv.lock and always upgrade the package to the latest
 upgrade-package = ["jax", "jaxlib", "orbax-checkpoint"]
+
+[tool.uv.sources]
+torch = { index = "pytorch" }
+
+[[tool.uv.index]]
+name = "pytorch"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true