test

Summary: kkkkkkkk Differential Revision: D36037141 fbshipit-source-id: 04d6ca49c813f52fc131eb526b0447163be2ce93
facebookresearch · May 4, 2022 · 17799b0 · 17799b0
1 parent d48968a
commit 17799b0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/reagent/mab/thompson_sampling.py b/reagent/mab/thompson_sampling.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
-
+# noqa
 from abc import abstractmethod
 from typing import Optional, List
 

diff --git a/setup.cfg b/setup.cfg
@@ -52,7 +52,9 @@ ax = ax-platform
 
 lite = nevergrad>=0.4.3
 
-torchrec_gpu = torchrec-nightly
+torchrec_gpu =
+  fbgemm-gpu-nightly
+  torchrec-nightly
 
 torchrec_cpu = torchrec-nightly-cpu