From 8889cf2d855d5b31cead345fc5de7461ad007a31 Mon Sep 17 00:00:00 2001 From: Zhengxing Chen Date: Wed, 4 May 2022 09:47:33 -0700 Subject: [PATCH 1/2] test Summary: kkkkkkkk Differential Revision: D36037141 fbshipit-source-id: 651ae7c02b3a04b247862ca99d6b4c30334af2e5 --- 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 From 31fc8d960bd7dc31ca121afd6ae579d977e13126 Mon Sep 17 00:00:00 2001 From: Shabab Ayub Date: Wed, 4 May 2022 09:47:53 -0700 Subject: [PATCH 2/2] fbgemm op ci test fail (#638) Summary: Pull Request resolved: https://github.com/facebookresearch/ReAgent/pull/638 Differential Revision: D36133994 fbshipit-source-id: d76a42b84f3eab4196a5d4f8210f3f37c5edf55e --- reagent/test/preprocessing/test_transforms.py | 1 + setup.cfg | 1 + 2 files changed, 2 insertions(+) diff --git a/reagent/test/preprocessing/test_transforms.py b/reagent/test/preprocessing/test_transforms.py index b63391487..57e9542d2 100644 --- a/reagent/test/preprocessing/test_transforms.py +++ b/reagent/test/preprocessing/test_transforms.py @@ -9,6 +9,7 @@ import numpy as np import reagent.core.types as rlt import torch +import fbgemm_gpu from reagent.preprocessing import transforms from reagent.preprocessing.types import InputColumn diff --git a/setup.cfg b/setup.cfg index 43b02a6d3..019b8804d 100644 --- a/setup.cfg +++ b/setup.cfg @@ -17,6 +17,7 @@ install_requires = # issue: https://github.com/openai/spinningup/issues/178 cloudpickle~=1.2.0 iopath + fbgemm-gpu-nightly numpy>=1.17.2 pandas>=1.0.3 # https://github.com/samuelcolvin/pydantic/issues/2042