fbgemm op ci test fail (#638)

Summary: Pull Request resolved: #638

Differential Revision: D36133994

fbshipit-source-id: d76a42b84f3eab4196a5d4f8210f3f37c5edf55e

Author: s4ayub

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
 

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