test (facebookresearch#636)

Summary: Pull Request resolved: facebookresearch#636 kkkkkkkk Differential Revision: D36037141 fbshipit-source-id: 254fe85c831b044a3ed58f6164e8b17d28d6c2d2
czxttkl · May 2, 2022 · 50d3db1 · 50d3db1
1 parent d48968a
commit 50d3db1
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