test

Summary: kkkkkkkk

Differential Revision: D36037141

fbshipit-source-id: 651ae7c02b3a04b247862ca99d6b4c30334af2e5

Author: czxttkl

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
 

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