Add missing dependency

2 files changed, 12 insertions, 5 deletions

diff --git a/.SRCINFO b/.SRCINFO
index c570a7eb1ad9..15786256d760 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,17 +1,18 @@
 pkgbase = fairseq-git
 	pkgdesc = Sequence modeling toolkit
 	pkgver = 0.11
-	pkgrel = 1
+	pkgrel = 2
 	url = https://github.com/pytorch/fairseq
 	arch = i686
 	arch = x86_64
 	license = MIT
-	depends = python-pytorch
 	depends = cuda
 	depends = nccl
-	depends = python-omegaconf
 	depends = python-antlr4
+	depends = python-bitarray
 	depends = python-hydra
+	depends = python-omegaconf
+	depends = python-pytorch
 	optdepends = python-apex: Faster training
 	provides = fairseq
 	conflicts = fairseq
diff --git a/PKGBUILD b/PKGBUILD
index 50b93916dfaa..d389e101613f 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,14 +2,20 @@
 
 pkgname=fairseq-git
 pkgver=0.11
-pkgrel=1
+pkgrel=2
 pkgdesc="Sequence modeling toolkit"
 arch=('i686' 'x86_64')
 url="https://github.com/pytorch/fairseq"
 license=('MIT')
 provides=('fairseq')
 conflicts=('fairseq')
-depends=('python-pytorch' 'cuda' 'nccl' 'python-omegaconf' 'python-antlr4' 'python-hydra')
+depends=('cuda'
+         'nccl'
+         'python-antlr4'
+         'python-bitarray'
+         'python-hydra'
+         'python-omegaconf'
+         'python-pytorch')
 optdepends=('python-apex: Faster training')
 source=("git+$url")
 sha256sums=('SKIP')