Fix dependency incompatibility in github tests (#207)

Summary: Pull Request resolved: #207 Differential Revision: D62472378
pytorch · Sep 11, 2024 · 4a5fa11 · 4a5fa11
1 parent a88a579
commit 4a5fa11
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/unit_test.yaml b/.github/workflows/unit_test.yaml
@@ -24,8 +24,7 @@ jobs:
         shell: bash -l {0}
         run: |
           set -eux
-          conda activate test
-          conda install pytorch torchaudio torchvision cpuonly -c pytorch-nightly
+          pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
           pip install -r requirements.txt
           pip install -r dev-requirements.txt
           pip install --no-build-isolation -e ".[dev]"

diff --git a/torcheval/metrics/aggregation/cov.py b/torcheval/metrics/aggregation/cov.py
@@ -7,14 +7,14 @@
 # pyre-strict
 
 from collections.abc import Iterable
-from typing import Tuple, TypeVar
+from typing import Tuple, TypeVar, Union
 
 import torch
 from torcheval.metrics.metric import Metric
 from typing_extensions import Self, TypeAlias
 
 # TODO: use a NamedTuple?
-_T = TypeVar("_T", bound=torch.Tensor | int)
+_T = TypeVar("_T", bound=Union[torch.Tensor, int])
 _Output: TypeAlias = Tuple[torch.Tensor, torch.Tensor]  # mean, cov