From 50d3db11253088f31743babf65df8394564433ed Mon Sep 17 00:00:00 2001
From: Zhengxing Chen <czxttkl@fb.com>
Date: Sun, 1 May 2022 20:08:32 -0700
Subject: [PATCH] test (#636)

Summary:
Pull Request resolved: https://github.com/facebookresearch/ReAgent/pull/636

kkkkkkkk

Differential Revision: D36037141

fbshipit-source-id: 254fe85c831b044a3ed58f6164e8b17d28d6c2d2
---
 reagent/mab/thompson_sampling.py | 2 +-
 setup.cfg                        | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/reagent/mab/thompson_sampling.py b/reagent/mab/thompson_sampling.py
index a61f1330f..72967d59b 100644
--- 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
index 7d3612167..43b02a6d3 100644
--- 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