Update tensor-regression dependency

Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Aug 7, 2024 · 84fcf6d · 84fcf6d
1 parent 4ee7fe4
commit 84fcf6d
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/pdm.lock b/pdm.lock
diff --git a/project/main_test.py b/project/main_test.py
@@ -30,7 +30,7 @@ def test_torch_can_use_the_GPU():
 @use_overrides([""])
 def test_defaults(experiment_config: Config) -> None:
     assert experiment_config.algorithm["_target_"] == (
-        ExampleAlgorithm.__module__ + "." + ExampleAlgorithm.__name__
+        ExampleAlgorithm.__module__ + "." + ExampleAlgorithm.__qualname__
     )
     assert (
         isinstance(experiment_config.datamodule, CIFAR10DataModule)

diff --git a/pyproject.toml b/pyproject.toml
@@ -27,7 +27,6 @@ dependencies = [
     "tensorboard>=2.16.2",
     "gymnax>=0.0.8",
     "torch-jax-interop>=0.0.7",
-    "tensor-regression>=0.0.4",
     "simple-parsing>=0.1.5",
     "pydantic==2.7.4",
     "milatools>=0.0.18",
@@ -61,7 +60,7 @@ dev = [
     "ruff>=0.3.3",
     "pytest-benchmark>=4.0.0",
     "pytest-cov>=5.0.0",
-    "tensor-regression @ git+https://www.github.com/lebrice/tensor_regression@add-skip-creation",
+    "tensor-regression>=0.0.6",
     "pytest-testmon>=2.1.1",
     "mkdocs>=1.6.0",
     "pymdown-extensions>=10.8.1",