From a292c0be6486d1e4f3997dce180ac44c91667088 Mon Sep 17 00:00:00 2001 From: kalyanvasudev Date: Fri, 12 Aug 2022 13:47:12 -0700 Subject: [PATCH 1/2] update python version in ci --- .circleci/config.yml | 2 +- pytorchvideo/layers/attention.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index df8133aa..e22b8cb3 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -56,7 +56,7 @@ setup_ptv_conda: &setup_ptv_conda conda update -y conda conda init bash source ~/.bashrc - conda create --name pytorchvideo python=3.7.9 + conda create --name pytorchvideo python=3.8.13 install_pytorch: &install_pytorch - run: diff --git a/pytorchvideo/layers/attention.py b/pytorchvideo/layers/attention.py index bfcf61ef..f6fdc739 100644 --- a/pytorchvideo/layers/attention.py +++ b/pytorchvideo/layers/attention.py @@ -4,6 +4,7 @@ import numpy import torch +import torch.fx import torch.nn as nn from torch.nn.common_types import _size_3_t From 3709ce4b1b24f00acbe822c9064934da623e362a Mon Sep 17 00:00:00 2001 From: kalyanvasudev Date: Fri, 12 Aug 2022 13:48:55 -0700 Subject: [PATCH 2/2] freeze av version --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index abe7eefc..b4fffca8 100755 --- a/setup.py +++ b/setup.py @@ -52,7 +52,7 @@ def get_name(): python_requires=">=3.7", install_requires=[ "fvcore", - "av", + "av==8.0.3", "parameterized", "iopath", "networkx",